Initial commit
This commit is contained in:
AlacrisDevs
132
scripts/add_stats_fields.py
Normal file
132
scripts/add_stats_fields.py
Normal file
@@ -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())
|
||||
76
scripts/migrate_to_pb.py
Normal file
76
scripts/migrate_to_pb.py
Normal file
@@ -0,0 +1,76 @@
|
||||
"""One-time migration: copy data/economy.json → PocketBase.
|
||||
|
||||
Run this ONCE after setting up PocketBase and before starting the bot with
|
||||
the new PocketBase backend.
|
||||
|
||||
Usage (run from project root):
|
||||
python scripts/migrate_to_pb.py
|
||||
|
||||
Requirements:
|
||||
- PocketBase must be running and reachable at PB_URL
|
||||
- PB_ADMIN_EMAIL and PB_ADMIN_PASSWORD must be set in .env
|
||||
- The 'economy_users' collection must already exist in PocketBase
|
||||
(see README or PocketBase admin UI to create it)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# Ensure the project root is on sys.path so pb_client can be imported
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||||
|
||||
load_dotenv()
|
||||
|
||||
import pb_client # noqa: E402 (needs dotenv loaded first)
|
||||
|
||||
DATA_FILE = Path("data") / "economy.json"
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
if not DATA_FILE.exists():
|
||||
print(f"[ERROR] {DATA_FILE} not found - nothing to migrate.")
|
||||
sys.exit(1)
|
||||
|
||||
raw: dict[str, dict] = json.loads(DATA_FILE.read_text(encoding="utf-8"))
|
||||
total = len(raw)
|
||||
print(f"Found {total} user(s) in {DATA_FILE}")
|
||||
|
||||
created = skipped = errors = 0
|
||||
|
||||
for uid, user in raw.items():
|
||||
try:
|
||||
record = dict(user)
|
||||
record["user_id"] = uid
|
||||
record.setdefault("balance", 0)
|
||||
record.setdefault("exp", 0)
|
||||
record.setdefault("items", [])
|
||||
record.setdefault("item_uses", {})
|
||||
record.setdefault("reminders", ["daily", "work", "beg", "crime", "rob"])
|
||||
record.setdefault("eco_banned", False)
|
||||
record.setdefault("daily_streak", 0)
|
||||
|
||||
existing = await pb_client.get_record(uid)
|
||||
if existing:
|
||||
await pb_client.update_record(existing["id"], record)
|
||||
print(f" [UPDATE] {uid}")
|
||||
skipped += 1 # reuse skipped counter as "updated"
|
||||
else:
|
||||
await pb_client.create_record(record)
|
||||
print(f" [CREATE] {uid}")
|
||||
created += 1
|
||||
except Exception as exc:
|
||||
print(f" [ERROR] {uid}: {exc}")
|
||||
errors += 1
|
||||
|
||||
print(f"\nDone. Created: {created} Skipped: {skipped} Errors: {errors}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user