updated to be simpler to modify the days input

Donald-rdex · Dec 4, 2023 · 53fad1b · 53fad1b
1 parent 8aeb728
commit 53fad1b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -6,6 +6,16 @@
 if __name__ == '__main__':
     today = '04'
 
+    input_file = f'problem_inputs/day{today}_input.txt'
+    # input_file = f'problem_inputs/test_input.txt'
+
+    with open(input_file) as day_fp:
+        day_input = day_fp.readlines()
+        day_input = [f'{i}'.strip() for i in day_input]
+        if today == '05':
+            day05.part01(day_input)
+            day05.part02(day_input)
+
     if today == '04':
         input_file = f'problem_inputs/day{today}_input.txt'
         # input_file = f'problem_inputs/test_input.txt'