feat: add new arguments

This commit is contained in:
Kristofers Solo 2024-01-03 17:03:31 +02:00
parent fc305cd239
commit 2787058392

26
main.py
View File

@ -2,6 +2,7 @@
import argparse import argparse
# from ai import train
from loguru import logger from loguru import logger
from utils import BASE_PATH from utils import BASE_PATH
@ -32,7 +33,9 @@ group.add_argument(
help="Verbose", help="Verbose",
) )
parser.add_argument( group2 = parser.add_mutually_exclusive_group()
group2.add_argument(
"-t", "-t",
"--train", "--train",
type=pos_int, type=pos_int,
@ -41,6 +44,19 @@ parser.add_argument(
metavar="int", metavar="int",
help="Trains the AI", help="Trains the AI",
) )
group2.add_argument(
"-n",
"--noui",
action="store_true",
help="Run app in the CLI",
)
group2.add_argument(
"-g",
"--graphic",
action="store_true",
help="Run app with GUI",
)
def setup_logger(debug_level: str) -> None: def setup_logger(debug_level: str) -> None:
@ -63,10 +79,12 @@ def main(args: argparse.ArgumentParser) -> None:
setup_logger("warning") setup_logger("warning")
if args.train is not None: if args.train is not None:
logger.warning(args.train) # train(args.train)
train(args.train) logger.debug("Train")
elif args.noui:
logger.debug("Run game in CLI")
else: else:
Menu().run() logger.debug("Run app")
if __name__ == "__main__": if __name__ == "__main__":