From 76f403a6ed8430064bfca6bbefa7753978717d6a Mon Sep 17 00:00:00 2001 From: pettai Date: Wed, 5 Mar 2025 16:04:57 +0100 Subject: [PATCH] load from config, don't hardcode --- .../overlay/usr/sbin/knot_lastlogin | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/dns-rest-api1.sunet.se/overlay/usr/sbin/knot_lastlogin b/dns-rest-api1.sunet.se/overlay/usr/sbin/knot_lastlogin index ede0a22..4606c05 100755 --- a/dns-rest-api1.sunet.se/overlay/usr/sbin/knot_lastlogin +++ b/dns-rest-api1.sunet.se/overlay/usr/sbin/knot_lastlogin @@ -1,9 +1,18 @@ #!/usr/bin/env python3 +import yaml import sqlite3 try: - with sqlite3.connect('/var/lib/knot_rest/database.db') as conn: + with open('/etc/knot_rest/knot_rest.yaml') as stream: + yamlconf = yaml.safe_load(stream) +except Exception as e: + print(e) + +knotrestdb = yamlconf["database"].removeprefix("sqlite:///") + +try: + with sqlite3.connect(knotrestdb) as conn: cur = conn.cursor() cur.execute('select username, description, logged_in from user') rows = cur.fetchall() @@ -14,7 +23,7 @@ try: description = "(no description)" lastlogin = row[2] if lastlogin == None: - lastlogin = "***never logged in***" + lastlogin = "**never logged in**" print(f"{username:<27}{description:<27}{lastlogin:<23}") except sqlite3.OperationalError as e: print(e)