From 469092002b7f1e1657468941bd86ccd3738baac3 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Thu, 1 Dec 2022 09:36:18 +0100 Subject: Added pre-commit --- aoc/__main__.py | 32 +++++++++++++++++++------------- 1 file changed, 19 insertions(+), 13 deletions(-) (limited to 'aoc/__main__.py') diff --git a/aoc/__main__.py b/aoc/__main__.py index 25a7026..337bdfe 100644 --- a/aoc/__main__.py +++ b/aoc/__main__.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- import argparse import importlib import os @@ -7,32 +8,37 @@ from typing import List def day(assignment_part: str) -> str: return { - '1': 'One', - '2': 'Two', + "1": "One", + "2": "Two", }[assignment_part] + def kwargs(kwarg: str) -> List: - return kwarg.split('=') + return kwarg.split("=") + -parser = argparse.ArgumentParser(description='Advent of Code') +parser = argparse.ArgumentParser(description="Advent of Code") -parser.add_argument('day', type=str, help='Assignment day') +parser.add_argument("day", type=str, help="Assignment day") parser.add_argument( - '-p', '--part', type=day, nargs='?', default='1', - help='Assingment part. Defaults to one.' + "-p", + "--part", + type=day, + nargs="?", + default="1", + help="Assingment part. Defaults to one.", ) -parser.add_argument('--example', default=False, action='store_true') -parser.add_argument('kwargs', type=kwargs, nargs='*') +parser.add_argument("--example", default=False, action="store_true") +parser.add_argument("kwargs", type=kwargs, nargs="*") -if __name__ == '__main__': +if __name__ == "__main__": args = parser.parse_args() assignment_day = importlib.import_module(args.day) - Assignment = getattr(assignment_day, f'Assignment{args.part}') + Assignment = getattr(assignment_day, f"Assignment{args.part}") assignment = Assignment( - path=os.path.dirname(assignment_day.__file__), - **dict(args.kwargs) + path=os.path.dirname(assignment_day.__file__), **dict(args.kwargs) ) start = perf_counter() -- cgit v1.2.3