Merge pull request #9 from jasonMPM/v7
Update app.py
This commit was merged in pull request #9.
This commit is contained in:
22
app.py
22
app.py
@@ -30,21 +30,25 @@ def get_db():
|
||||
|
||||
def init_db():
|
||||
with get_db() as db:
|
||||
db.execute("""
|
||||
db.execute(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS badge_events (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
actor_id TEXT NOT NULL,
|
||||
ts TEXT NOT NULL,
|
||||
date TEXT NOT NULL
|
||||
)
|
||||
""")
|
||||
db.execute("""
|
||||
"""
|
||||
)
|
||||
db.execute(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS user_cache (
|
||||
actor_id TEXT PRIMARY KEY,
|
||||
full_name TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL
|
||||
)
|
||||
""")
|
||||
"""
|
||||
)
|
||||
db.commit()
|
||||
|
||||
|
||||
@@ -62,13 +66,12 @@ def sync_unifi_users():
|
||||
users = r.json().get("data", [])
|
||||
with get_db() as db:
|
||||
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 = (
|
||||
u.get("id") or
|
||||
u.get("identity_id") or
|
||||
u.get("user_id")
|
||||
u.get("id")
|
||||
or u.get("identity_id")
|
||||
or u.get("user_id")
|
||||
)
|
||||
|
||||
if not actor_id:
|
||||
continue # skip malformed entries
|
||||
|
||||
@@ -90,7 +93,6 @@ def sync_unifi_users():
|
||||
datetime.utcnow().isoformat(),
|
||||
),
|
||||
)
|
||||
|
||||
db.commit()
|
||||
log.info("Synced %d users from UniFi Access", len(users))
|
||||
except Exception as e:
|
||||
|
||||
Reference in New Issue
Block a user