accept input file as cmdline argument

2021/day1/day1.py

@@ -40,5 +40,7 @@ def part2(infile):
 
 
 if __name__ == "__main__":
-    part1("./input.txt")
-    part2("./input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    part1(infile)
+    part2(infile)

2021/day2/day2.py

@@ -39,4 +39,6 @@ def main(input_file):
 
 
 if __name__ == "__main__":
-    main("input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    main(infile)

2021/day3/day3.py

@@ -64,4 +64,6 @@ def main(input_file):
 
 
 if __name__ == "__main__":
-    main("input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    main(infile)

2021/day4/day4.py

@@ -94,4 +94,6 @@ def main(input_file):
 
 
 if __name__ == "__main__":
-    main("input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    main(infile)

2021/day5/day5.py

@@ -35,7 +35,7 @@ def main(infile):
     print(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.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)}')

2021/day7/day7.py

@@ -26,8 +26,10 @@ def part2(crabs):
 
 
 if __name__ == "__main__":
-    with open("input.txt") as infile:
-        crabs = [int(x) for x in infile.readline().split(",")]
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    with open(infile) as f:
+        crabs = [int(x) for x in f.readline().split(",")]
     part1(crabs)
     part2(crabs)
 

2021/day8/day8.py

@@ -43,4 +43,6 @@ def main(infile):
 
 
 if __name__ == "__main__":
-    main("input.txt")
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    main(infile)