Add CLI option to fetch variants

This commit is contained in:
Maximilian Keßler 2023-11-22 18:58:55 +01:00
parent dddb9de878
commit 01ebfa380d
Signed by: max
GPG key ID: BCC5A619923C0BA5
2 changed files with 30 additions and 27 deletions

View file

@ -72,6 +72,32 @@ def init_database():
logger.verbose("Initialized DB tables.")
def fetch_and_initialize_variants():
response = requests.get(constants.VARIANTS_JSON_URL)
if not response.status_code == 200:
logger.error("Could not download variants.json file from github (tried url {})".format(constants.VARIANTS_JSON_URL))
return
variants = json.loads(response.text)
config = read_config()
for variant in variants:
variant_id = variant['id']
name = variant['name']
clue_starved = variant.get('clueStarved', False)
num_suits = len(variant['suits'])
if config.min_suit_count <= num_suits <= config.max_suit_count:
if any(var_name in name.lower() for var_name in config.excluded_variants):
continue
cur = conn_manager.get_new_cursor()
cur.execute(
"INSERT INTO variants (id, name, num_suits, clue_starved) VALUES (%s, %s, %s, %s)",
(variant_id, name, num_suits, clue_starved)
)
conn_manager.get_connection().commit()
def add_player_name(player_name: str):
conn = conn_manager.get_connection()
cur = conn.cursor()
@ -121,29 +147,3 @@ def add_user_name_to_player(hanabi_username: str, player_name: str):
def add_player(player_name: str, user_name: str):
add_player_name(player_name)
add_user_name_to_player(user_name, player_name)
def fetch_and_initialize_variants():
response = requests.get(constants.VARIANTS_JSON_URL)
if not response.status_code == 200:
logger.error("Could not download variants.json file from github (tried url {})".format(constants.VARIANTS_JSON_URL))
return
variants = json.loads(response.text)
config = read_config()
for variant in variants:
variant_id = variant['id']
name = variant['name']
clue_starved = variant.get('clueStarved', False)
num_suits = len(variant['suits'])
if config.min_suit_count <= num_suits <= config.max_suit_count:
if any(var_name in name.lower() for var_name in config.excluded_variants):
continue
cur = conn_manager.get_new_cursor()
cur.execute(
"INSERT INTO variants (id, name, num_suits, clue_starved) VALUES (%s, %s, %s, %s)",
(variant_id, name, num_suits, clue_starved)
)
conn_manager.get_connection().commit()

View file

@ -11,7 +11,7 @@ import log_setup
from log_setup import logger
def subcommand_init(force: bool):
def subcommand_init(force: bool, fetch_variants: bool):
tables = database.get_existing_tables()
if len(tables) > 0 and not force:
logger.info(
@ -29,6 +29,8 @@ def subcommand_init(force: bool):
return
database.init_database()
logger.info("Successfully initialized database tables")
if fetch_variants:
database.fetch_and_initialize_variants()
def subcommand_generate_config():
@ -47,6 +49,7 @@ def get_parser() -> argparse.ArgumentParser:
init_parser = subparsers.add_parser('init', help='Initialize database.')
init_parser.add_argument('--force', '-f', help='Force initialization (Drops existing tables)', action='store_true')
init_parser.add_argument('--fetch-variants', '-v', help='Fetch and initialize variants', action='store_true')
subparsers.add_parser('generate-config', help='Generate config file at default location')