refactor: split database.py into package (ref.md 3.2)

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

devplacepy/database/users.py

@@ -0,0 +1,84 @@
+# retoor <retoor@molodetz.nl>
+
+from .core import TTLCache, bump_cache_version, db, sync_local_cache
+
+
+def get_users_by_uids(uids):
+    if not uids or "users" not in db.tables:
+        return {}
+    users = db["users"]
+    if "uid" not in users.columns:
+        return {}
+    seen = set()
+    unique = [u for u in uids if u not in seen and not seen.add(u)]
+    return {u["uid"]: u for u in users.find(users.table.columns.uid.in_(unique))}
+
+
+_admins_cache = TTLCache(ttl=300, max_size=4)
+
+
+def invalidate_admins_cache() -> None:
+    _admins_cache.clear()
+    bump_cache_version("admins")
+
+
+def get_admin_uids():
+    sync_local_cache("admins", _admins_cache)
+    cached = _admins_cache.get("uids")
+    if cached is not None:
+        return list(cached)
+    if "users" not in db.tables:
+        return []
+    rows = db.query("SELECT uid FROM users WHERE role = 'Admin'")
+    uids = [row["uid"] for row in rows]
+    _admins_cache.set("uids", uids)
+    return list(uids)
+
+
+def set_user_timezone(user_uid: str, tz_name: str) -> None:
+    if "users" not in db.tables or not user_uid or not tz_name:
+        return
+    users = db["users"]
+    if not users.has_column("timezone"):
+        users.create_column_by_example("timezone", "")
+    current = users.find_one(uid=user_uid)
+    if current and current.get("timezone") == tz_name:
+        return
+    users.update({"uid": user_uid, "timezone": tz_name}, ["uid"])
+
+
+def get_primary_admin_uid():
+    sync_local_cache("admins", _admins_cache)
+    cached = _admins_cache.get("primary")
+    if cached is not None:
+        return cached or None
+    if "users" not in db.tables:
+        return None
+    rows = list(
+        db.query(
+            "SELECT uid FROM users WHERE role = 'Admin' "
+            "ORDER BY created_at ASC, id ASC LIMIT 1"
+        )
+    )
+    primary = rows[0]["uid"] if rows else None
+    _admins_cache.set("primary", primary or "")
+    return primary
+
+
+def search_users_by_username(q, *, exclude_uid=None, limit=10):
+    if not q or "users" not in db.tables:
+        return []
+    if exclude_uid is not None:
+        rows = db.query(
+            "SELECT uid, username FROM users WHERE username LIKE :q AND uid != :me LIMIT :limit",
+            q=f"%{q}%",
+            me=exclude_uid,
+            limit=limit,
+        )
+    else:
+        rows = db.query(
+            "SELECT uid, username FROM users WHERE username LIKE :q LIMIT :limit",
+            q=f"%{q}%",
+            limit=limit,
+        )
+    return [{"uid": r["uid"], "username": r["username"]} for r in rows]