accept input file as cmdline argument

2018/day1/day1.py

@@ -17,6 +17,8 @@ def main(inp):
 
     
 if __name__ == '__main__':
-    with open("input.txt") as inp:
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    with open(infile) as inp:
         main(inp.readlines())
 

2019/day1/day1.py

@@ -32,10 +32,13 @@ def test_part2():
 
 
 if __name__ == "__main__":
-    print("Part 1 - total mass: ", calculate_total_fuel_mass("./input.txt"))
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+
+    print("Part 1 - total mass: ", calculate_total_fuel_mass(infile))
 
     test_part2()
     print(
         "Part 2 -- total mass: ",
-        calculate_total_fuel_mass("./input.txt", calculate_fuel_iterative),
+        calculate_total_fuel_mass(infile, calculate_fuel_iterative),
     )

2019/day2/day2.py

@@ -54,7 +54,9 @@ def run_program(memory, noun, verb):
 
 if __name__ == "__main__":
     tests()
-    with open("input.txt") as inp:
+    import sys
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    with open(infile) as inp:
         memory = [int(x) for x in inp.readline().strip().split(",")]
         # Pass a copy to avoid modifying the original memory
         print("Part 1 answer: ", run_program(memory[:], 12, 2))

2019/day3/day3.py

@@ -70,8 +70,9 @@ def tests():
 
 if __name__ == "__main__":
     tests()
-
+    import sys
-    with open("input.txt") as raw_input:
+    infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
+    with open(infile) as raw_input:
         lines = raw_input.readlines()
         wire1, wire2 = [line.strip("\n").split(",") for line in lines]
         print("Part 1 -- distance = ", find_min_distance(wire1, wire2))

2021/day1/day1.py

@@ -40,5 +40,7 @@ def part2(infile):
 
 
 if __name__ == "__main__":
-    part1("./input.txt")
+    import sys
-    part2("./input.txt")
+    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/day7/day7.py

@@ -26,8 +26,10 @@ def part2(crabs):
 
 
 if __name__ == "__main__":
-    with open("input.txt") as infile:
+    import sys
-        crabs = [int(x) for x in infile.readline().split(",")]
+    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)