NotSoToothless
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
"""Backfill tss_tournaments.db from tournament ids already in match_summary.
|
|
|
|
Usage:
|
|
python TSSBOT/scripts/backfill_tournaments.py [--dry-run] [--limit N] [--sleep SECONDS]
|
|
"""
|
|
import argparse
|
|
import asyncio
|
|
import pathlib
|
|
import sqlite3
|
|
import sys
|
|
from typing import List, Optional, Tuple
|
|
|
|
ROOT = pathlib.Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(ROOT))
|
|
sys.path.insert(0, str(ROOT.parent / "SHARED"))
|
|
|
|
from BOT.storage import TSS_BATTLES_DB_PATH # noqa: E402
|
|
from BOT.tss_tournaments import init_tss_tournaments_db, scan_and_store # noqa: E402
|
|
|
|
|
|
async def tournament_ids(limit: Optional[int]) -> List[Tuple[int, Optional[str]]]:
|
|
sql = """
|
|
SELECT tournament_id, NULLIF(MAX(tournament_name), '') AS tournament_name
|
|
FROM match_summary
|
|
WHERE tournament_id IS NOT NULL AND tournament_id > 0
|
|
GROUP BY tournament_id
|
|
ORDER BY MAX(endtime_unix) DESC
|
|
"""
|
|
if limit:
|
|
sql += " LIMIT ?"
|
|
params = (limit,)
|
|
else:
|
|
params = ()
|
|
|
|
with sqlite3.connect(TSS_BATTLES_DB_PATH) as conn:
|
|
rows = conn.execute(sql, params).fetchall()
|
|
return [(int(row[0]), row[1]) for row in rows]
|
|
|
|
|
|
async def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--dry-run", action="store_true")
|
|
parser.add_argument("--limit", type=int, default=None)
|
|
parser.add_argument("--sleep", type=float, default=1.0)
|
|
args = parser.parse_args()
|
|
|
|
rows = await tournament_ids(args.limit)
|
|
print(f"Found {len(rows)} tournament ids in {TSS_BATTLES_DB_PATH}")
|
|
if args.dry_run:
|
|
for tid, name in rows:
|
|
print(f" {tid}: {name or 'Tournament ' + str(tid)}")
|
|
return
|
|
|
|
await init_tss_tournaments_db()
|
|
for index, (tid, name) in enumerate(rows, start=1):
|
|
print(f"[{index}/{len(rows)}] scanning tournament {tid}")
|
|
await scan_and_store(tid, fallback_name=name)
|
|
if args.sleep and index < len(rows):
|
|
await asyncio.sleep(args.sleep)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|