accept input file as cmdline argument

2021/day6/day6.py

@@ -23,4 +23,6 @@ def main(infile):
     print(f"Part 2, {res}")
 
 if __name__ == "__main__":
-    main("input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    main(infile)

2021/day6/day6_2.py

@@ -31,4 +31,4 @@ def rotate(l):
 for j in range(256):
     numbers = rotate(numbers)
     numbers[6] += numbers[8]
-    print(f'DAY {j+1} AMOUNT OF FISH: {sum(numbers)}')
+    print(f'DAY {j+1} AMOUNT OF FISH: {sum(numbers)}')