Py-Hanabi/hanabi/hanabi_cli.py

85 lines
2.5 KiB
Python
Raw Normal View History

2023-05-14 19:11:51 +02:00
#! /usr/bin/python3
import argparse
import verboselogs
2023-05-14 19:11:51 +02:00
from hanabi import logger, logger_manager
from hanabi.live import check_game
from hanabi.live import download_data
from hanabi.live import compress
2023-05-14 19:11:51 +02:00
"""
init db + populate tables
download games of variant
download single game
analyze single game
"""
def add_init_subparser(subparsers):
parser = subparsers.add_parser(
'init',
help='Init database tables, retrieve variant and suit information from hanab.live'
)
def add_download_subparser(subparsers):
parser = subparsers.add_parser('download', help='Download games from hanab.live')
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--var', '-v', type=int)
group.add_argument('--id', '-i', type=int)
def add_analyze_subparser(subparsers):
parser = subparsers.add_parser('analyze', help='Analyze a game and find the last winning state')
parser.add_argument('game_id', type=int)
parser.add_argument('--download', '-d', help='Download game if not in database', action='store_true')
def analyze_game(game_id: int, download: bool = False):
if download:
download_data.detailed_export_game(game_id)
logger.info('Analyzing game {}'.format(game_id))
turn, sol = check_game.check_game(game_id)
2023-05-14 19:11:51 +02:00
if turn == 0:
logger.info('Instance is unfeasible')
2023-05-14 19:11:51 +02:00
else:
logger.info('Game was first lost after {} turns.'.format(turn))
logger.info(
'A replay achieving perfect score from the previous turn onwards is: {}#{}'
.format(compress.link(sol), turn)
2023-05-14 19:11:51 +02:00
)
def main_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(
prog='hanabi_suite',
description='High-level interface for analysis of hanabi instances.'
)
parser.add_argument('--verbose', '-v', help='Enable verbose logging to console', action='store_true')
2023-05-14 19:11:51 +02:00
subparsers = parser.add_subparsers(dest='command', required=True, help='select subcommand')
add_init_subparser(subparsers)
add_analyze_subparser(subparsers)
add_download_subparser(subparsers)
return parser
2023-07-04 22:19:13 +02:00
def hanabi_cli():
2023-05-14 19:11:51 +02:00
args = main_parser().parse_args()
switcher = {
'analyze': analyze_game
}
if args.verbose:
logger_manager.set_console_level(verboselogs.VERBOSE)
2023-05-14 19:11:51 +02:00
method_args = dict(vars(args))
method_args.pop('command')
method_args.pop('verbose')
2023-05-14 19:11:51 +02:00
switcher[args.command](**method_args)
if __name__ == "__main__":
2023-07-04 22:19:13 +02:00
hanabi_cli()