Solution to day three, parts one and two

Author: jda5

01/main.py

@@ -1,7 +1,7 @@
 from collections import Counter
-from utils import load_input
+from utils import load_input_lines
 
-location_ids = load_input('01')
+location_ids = load_input_lines('01')
 
 def puzzle_one() -> int:
     left, right = [], []

02/main.py

@@ -1,6 +1,6 @@
-from utils import load_input
+from utils import load_input_lines
 
-reports = load_input('02')
+reports = load_input_lines('02')
 
 
 def is_safe(a: int, b: int, is_increasing: bool) -> bool:

03/main.py

@@ -0,0 +1,37 @@
+import re
+from utils import load_input_full
+
+input_program = load_input_full('03')
+
+
+def puzzle_one(program: str) -> int:
+    groups = re.findall(r"mul\((\d+),(\d+)\)", program)
+    return sum(int(group[0]) * int(group[1]) for group in groups)
+
+def find_conditional(conditional: str, start: int) -> int:
+    try:
+        return input_program.index(conditional, start)
+    except ValueError:
+        return len(input_program)
+
+def puzzle_two() -> int:
+    enabled = True
+    start = 0
+    res = 0
+
+    while start < len(input_program):
+
+        if enabled:
+            end = find_conditional("don't()", start)
+            res += puzzle_one(input_program[start:end])
+            start = end + len("don't()")
+            enabled = False
+        else:
+            start = find_conditional("do()", start) + len("do()")
+            enabled = True
+
+    return res
+
+if __name__ == "__main__":
+    print(puzzle_one(program=input_program))
+    print(puzzle_two())

utils.py

@@ -2,7 +2,13 @@
 
 TOP = dirname(abspath(__file__))
 
-def load_input(puzzle_number: str) -> list[str]:
+
+def load_input_lines(puzzle_number: str) -> list[str]:
+    filepath = join(TOP, puzzle_number, 'input.txt')
+    with open(filepath) as file:
+        return file.read().splitlines()
+    
+def load_input_full(puzzle_number: str) -> str:
     filepath = join(TOP, puzzle_number, 'input.txt')
     with open(filepath) as file:
-        return file.read().splitlines()
+        return file.read()