Initial commit

scripts/add_stats_fields.py

@@ -0,0 +1,132 @@
+"""Add new schema fields to the economy_users PocketBase collection.
+
+Run once after pulling the stats + jailbreak_used changes:
+    python scripts/add_stats_fields.py
+
+Requirements:
+    - PocketBase running and reachable at PB_URL
+    - PB_ADMIN_EMAIL / PB_ADMIN_PASSWORD set in .env
+"""
+
+from __future__ import annotations
+
+import asyncio
+import os
+import sys
+from pathlib import Path
+
+import aiohttp
+from dotenv import load_dotenv
+
+sys.path.insert(0, str(Path(__file__).parent.parent))
+load_dotenv()
+
+PB_URL            = os.getenv("PB_URL", "http://127.0.0.1:8090")
+PB_ADMIN_EMAIL    = os.getenv("PB_ADMIN_EMAIL", "")
+PB_ADMIN_PASSWORD = os.getenv("PB_ADMIN_PASSWORD", "")
+COLLECTION        = "economy_users"
+
+# ---------------------------------------------------------------------------
+# New fields to add
+# ---------------------------------------------------------------------------
+_NEW_BOOL_FIELDS = [
+    "jailbreak_used",
+]
+
+_NEW_NUMBER_FIELDS = [
+    "heist_global_cd_until",
+    "peak_balance",
+    "lifetime_earned",
+    "lifetime_lost",
+    "work_count",
+    "beg_count",
+    "total_wagered",
+    "biggest_win",
+    "biggest_loss",
+    "slots_jackpots",
+    "crimes_attempted",
+    "crimes_succeeded",
+    "times_jailed",
+    "total_bail_paid",
+    "heists_joined",
+    "heists_won",
+    "total_given",
+    "total_received",
+    "best_daily_streak",
+]
+
+
+def _number_field(name: str) -> dict:
+    return {
+        "name": name,
+        "type": "number",
+        "required": False,
+        "options": {"min": None, "max": None, "noDecimal": True},
+    }
+
+
+def _bool_field(name: str) -> dict:
+    return {"name": name, "type": "bool", "required": False}
+
+
+async def main() -> None:
+    timeout = aiohttp.ClientTimeout(total=15)
+    async with aiohttp.ClientSession(timeout=timeout) as session:
+        # ── Authenticate ────────────────────────────────────────────────────
+        async with session.post(
+            f"{PB_URL}/api/collections/_superusers/auth-with-password",
+            json={"identity": PB_ADMIN_EMAIL, "password": PB_ADMIN_PASSWORD},
+        ) as resp:
+            if resp.status != 200:
+                print(f"Auth failed ({resp.status}): {await resp.text()}")
+                return
+            token = (await resp.json())["token"]
+
+        hdrs = {"Authorization": token}
+
+        # ── Fetch current collection ─────────────────────────────────────────
+        async with session.get(
+            f"{PB_URL}/api/collections/{COLLECTION}", headers=hdrs
+        ) as resp:
+            if resp.status != 200:
+                print(f"Could not fetch collection ({resp.status}): {await resp.text()}")
+                return
+            col = await resp.json()
+
+        existing = {f["name"] for f in col.get("fields", [])}
+        print(f"Existing fields ({len(existing)}): {sorted(existing)}\n")
+
+        new_fields = []
+        for name in _NEW_BOOL_FIELDS:
+            if name not in existing:
+                new_fields.append(_bool_field(name))
+                print(f"  + {name}  (bool)")
+            else:
+                print(f"  = {name}  (already exists)")
+
+        for name in _NEW_NUMBER_FIELDS:
+            if name not in existing:
+                new_fields.append(_number_field(name))
+                print(f"  + {name}  (number)")
+            else:
+                print(f"  = {name}  (already exists)")
+
+        if not new_fields:
+            print("\nNothing to add - schema already up to date.")
+            return
+
+        # ── Patch collection schema ──────────────────────────────────────────
+        updated_fields = col.get("fields", []) + new_fields
+        async with session.patch(
+            f"{PB_URL}/api/collections/{COLLECTION}",
+            json={"fields": updated_fields},
+            headers=hdrs,
+        ) as resp:
+            if resp.status != 200:
+                print(f"\nSchema update failed ({resp.status}): {await resp.text()}")
+                return
+            print(f"\n✅ Added {len(new_fields)} field(s) successfully.")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())