feat: better state_handler
This commit is contained in:
@@ -1,9 +1,7 @@
|
||||
from diceplayer.config.player_config import PlayerConfig
|
||||
from diceplayer.cli import ArgsModel, read_input
|
||||
from diceplayer.logger import logger
|
||||
from diceplayer.player import Player
|
||||
|
||||
import yaml
|
||||
|
||||
import argparse
|
||||
from importlib import metadata
|
||||
|
||||
@@ -11,17 +9,6 @@ from importlib import metadata
|
||||
VERSION = metadata.version("diceplayer")
|
||||
|
||||
|
||||
def read_input(infile) -> PlayerConfig:
|
||||
try:
|
||||
with open(infile, "r") as f:
|
||||
return PlayerConfig.model_validate(
|
||||
yaml.safe_load(f)
|
||||
)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to read input file")
|
||||
raise e
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(prog="Diceplayer")
|
||||
parser.add_argument(
|
||||
@@ -46,7 +33,7 @@ def main():
|
||||
metavar="OUTFILE",
|
||||
help="output file of diceplayer [default = run.log]",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
args = ArgsModel.from_args(parser.parse_args())
|
||||
|
||||
logger.set_output_file(args.outfile)
|
||||
|
||||
@@ -56,4 +43,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user