Update app.py #9

Merged
jasonMPM merged 1 commits from v7 into main 2026-03-04 22:17:12 -06:00

22
app.py
View File

@@ -30,21 +30,25 @@ def get_db():
def init_db(): def init_db():
with get_db() as db: with get_db() as db:
db.execute(""" db.execute(
"""
CREATE TABLE IF NOT EXISTS badge_events ( CREATE TABLE IF NOT EXISTS badge_events (
id INTEGER PRIMARY KEY AUTOINCREMENT, id INTEGER PRIMARY KEY AUTOINCREMENT,
actor_id TEXT NOT NULL, actor_id TEXT NOT NULL,
ts TEXT NOT NULL, ts TEXT NOT NULL,
date TEXT NOT NULL date TEXT NOT NULL
) )
""") """
db.execute(""" )
db.execute(
"""
CREATE TABLE IF NOT EXISTS user_cache ( CREATE TABLE IF NOT EXISTS user_cache (
actor_id TEXT PRIMARY KEY, actor_id TEXT PRIMARY KEY,
full_name TEXT NOT NULL, full_name TEXT NOT NULL,
updated_at TEXT NOT NULL updated_at TEXT NOT NULL
) )
""") """
)
db.commit() db.commit()
@@ -62,13 +66,12 @@ def sync_unifi_users():
users = r.json().get("data", []) users = r.json().get("data", [])
with get_db() as db: with get_db() as db:
for u in users: for u in users:
# Prefer the same ID used in webhooks (identity ID) # Prefer the same ID used in webhooks (identity ID / id)
actor_id = ( actor_id = (
u.get("id") or u.get("id")
u.get("identity_id") or or u.get("identity_id")
u.get("user_id") or u.get("user_id")
) )
if not actor_id: if not actor_id:
continue # skip malformed entries continue # skip malformed entries
@@ -90,7 +93,6 @@ def sync_unifi_users():
datetime.utcnow().isoformat(), datetime.utcnow().isoformat(),
), ),
) )
db.commit() db.commit()
log.info("Synced %d users from UniFi Access", len(users)) log.info("Synced %d users from UniFi Access", len(users))
except Exception as e: except Exception as e: