hanabi-league/main.py

#! python3

import argparse
import sys
import verboselogs


sys.path.append('src/')

import config
import constants
import database
import log_setup
import fetch_players
import fetch_games

from log_setup import logger


def subcommand_init(force: bool, no_fetch_variants: bool):
    tables = database.get_existing_tables()
    if len(tables) > 0 and not force:
        logger.info(
            'Database tables "{}" exist already, aborting. To force re-initialization, use the --force options'
            .format(", ".join(tables))
        )
        return
    if len(tables) > 0:
        logger.info(
            "WARNING: This will drop all existing tables from the database and re-initialize them."
        )
        response = input("Do you wish to continue? [y/N] ")
        if response not in ["y", "Y", "yes"]:
            logger.info("Database initialization aborted.")
            return
    database.init_database()
    logger.info("Successfully initialized database tables")
    if not no_fetch_variants:
        database.fetch_and_initialize_variants()
        database.initialize_variant_base_ratings()
        logger.info("Successfully initialized variants and base ratings.")


def subcommand_generate_config():
    config.create_db_config()
    config.create_config()


def subcommand_fetch(target: str):
    if target in ["all", "players"]:
        fetch_players.fetch_players_interactive()
    if target in ["all", "games"]:
        games = fetch_games.fetch_games_for_all_players()
        fetch_games.store_new_games(games)
    if target in ["all", "game-details"]:
        fetch_games.fetch_all_game_details()


def get_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        prog=constants.APP_NAME,
        description="Data collection and analysis tool for the Hanabi League"
    )
    parser.add_argument('--verbose', '-v', help='Enable verbose logging to console', action='store_true')

    subparsers = parser.add_subparsers(dest='command', required=True, help='select subcommand')

    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('--no-fetch-variants', '-n', help='Do not fetch and initialize variants', action='store_true')

    subparsers.add_parser('generate-config', help='Generate config file at default location')

    fetch_parser = subparsers.add_parser('fetch', help='Fetch new data.')
    fetch_parser.add_argument(dest='target', choices=['all', 'players', 'games', 'game-details'], default='all')

    return parser


def main():
    parser = get_parser()
    args = parser.parse_args()

    subcommand_func = {
        'init': subcommand_init,
        'generate-config': subcommand_generate_config,
        'fetch': subcommand_fetch
    }[args.command]

    if args.verbose:
        log_setup.logger_manager.set_console_level(verboselogs.VERBOSE)

    del args.command
    del args.verbose

    subcommand_func(**vars(args))


if __name__ == "__main__":
    main()