diff options
| -rw-r--r-- | 20/py/aoc20.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/20/py/aoc20.py b/20/py/aoc20.py index 9380915..101ddb0 100644 --- a/20/py/aoc20.py +++ b/20/py/aoc20.py @@ -28,12 +28,14 @@ if __name__ == "__main__": only = set() only_part = 0 run_times = 1 + input_root = "../input" argv, argc = sys.argv, len(sys.argv) i = 1 while i < argc: if argv[i] == "--help": - print(f"usage: {argv[0]} [--help] [--time [times]] [--skip <n> <n> ...] [--only <n> <n> ...] [--part 0|1|2]") + print(f"usage: {argv[0]} [--help] [--time [times]] [--skip <n> <n> ...]\n" + + " [--only <n> <n> ...] [--part 0|1|2] [--input <dir>]") sys.exit(0) elif argv[i] == "--time": i += 1 @@ -56,6 +58,10 @@ if __name__ == "__main__": i += 1 only_part = int(argv[i]) i += 1 + elif argv[i] == "--input": + i += 1 + input_root = argv[i] + i += 1 else: print(f"unknown argument {argv[i]}") print(f"maybe try {argv[0]} --help ?") @@ -70,7 +76,7 @@ if __name__ == "__main__": d06, d07, d08, d09)): if day+1 in skip or (only and day+1 not in only): continue - input = open(f"../input/{day+1:02}").readlines() + input = open(f"{input_root}/{day+1:02}").readlines() for part, part_func in enumerate((mod.pt1, mod.pt2)): if only_part != 0 and part+1 != only_part: continue |