mirror of
https://github.com/evroon/bracket.git
synced 2026-02-01 10:01:16 -05:00
78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
import asyncio
|
|
import functools
|
|
from typing import Any
|
|
|
|
import click
|
|
from sqlalchemy import Table
|
|
|
|
from ladderz.database import database, engine
|
|
from ladderz.logger import get_logger
|
|
from ladderz.schema import clubs, matches, metadata, players, rounds, teams, tournaments, users
|
|
from ladderz.utils.dummy_records import (
|
|
DUMMY_CLUBS,
|
|
DUMMY_MATCHES,
|
|
DUMMY_PLAYERS,
|
|
DUMMY_ROUNDS,
|
|
DUMMY_TEAMS,
|
|
DUMMY_TOURNAMENTS,
|
|
DUMMY_USERS,
|
|
)
|
|
from ladderz.utils.types import BaseModelT
|
|
|
|
logger = get_logger('cli')
|
|
|
|
|
|
def run_async(f: Any) -> Any:
|
|
@functools.wraps(f)
|
|
def wrapper(*args: Any, **kwargs: Any) -> Any:
|
|
loop = asyncio.new_event_loop()
|
|
|
|
async def inner() -> None:
|
|
try:
|
|
await database.connect()
|
|
await f(*args, **kwargs)
|
|
|
|
except KeyboardInterrupt:
|
|
logger.debug('Closing the process.')
|
|
except Exception as e:
|
|
logger.error(e, exc_info=True)
|
|
raise e
|
|
finally:
|
|
await database.disconnect()
|
|
|
|
return loop.run_until_complete(inner())
|
|
|
|
return wrapper
|
|
|
|
|
|
@click.group()
|
|
def cli() -> None:
|
|
pass
|
|
|
|
|
|
async def bulk_insert(table: Table, rows: list[BaseModelT]) -> None:
|
|
for row in rows:
|
|
await database.execute(query=table.insert(), values=row.dict())
|
|
|
|
|
|
@click.command()
|
|
@run_async
|
|
async def create_dev_db() -> None:
|
|
logger.warning('Initializing database with dummy records')
|
|
await database.connect()
|
|
metadata.drop_all(engine)
|
|
metadata.create_all(engine)
|
|
|
|
await bulk_insert(users, DUMMY_USERS)
|
|
await bulk_insert(clubs, DUMMY_CLUBS)
|
|
await bulk_insert(tournaments, DUMMY_TOURNAMENTS)
|
|
await bulk_insert(teams, DUMMY_TEAMS)
|
|
await bulk_insert(players, DUMMY_PLAYERS)
|
|
await bulk_insert(rounds, DUMMY_ROUNDS)
|
|
await bulk_insert(matches, DUMMY_MATCHES)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli.add_command(create_dev_db)
|
|
cli()
|