[py] Port 2018_01

py/2018/01/solve.py

@@ -1,33 +0,0 @@
-import sys
-
-def load_freqs(filename):
-	freqs = []
-	with open(filename, "r") as f:
-		for line in f:
-			n = int(line[:-1])
-			freqs.append(n)
-	return freqs
-
-# PART 2
-
-def find_repeat(freqs):
-	total = 0
-	found = {total}
-
-	while True:
-		for n in freqs:
-			total += n
-			if total in found:
-				return total
-			else:
-				found.add(total)
-
-def main(filename):
-	freqs = load_freqs(filename)
-	print(f"Solutions for {filename}")
-	print(f"Part 1: {sum(freqs)}")
-	print(f"Part 2: {find_repeat(freqs)}")
-
-if __name__ == "__main__":
-	for filename in sys.argv[1:]:
-		main(filename)

py/2018/01/test_input.txt

@@ -1,5 +0,0 @@
-+1
-+2
--3
--4
-+5

py/aoc/__init__.py

@@ -2,11 +2,13 @@ import sys
 import argparse
 from pathlib import Path
 
+from .y2018 import d01
 from .y2020 import d10
 from .y2021 import d14
 from .y2022 import d01, d02, d03, d04, d05, d06
 
 DAYS = {
+    "2018_01": y2018.d01.solve,
     "2020_10": y2020.d10.solve,
     "2021_14": y2021.d14.solve,
     "2022_01": y2022.d01.solve,

py/aoc/y2018/d01.py

@@ -0,0 +1,17 @@
+def find_repeat(freqs):
+    total = 0
+    found = {total}
+
+    while True:
+        for n in freqs:
+            total += n
+            if total in found:
+                return total
+            else:
+                found.add(total)
+
+
+def solve(inputstr):
+    freqs = [int(freq) for freq in inputstr.splitlines()]
+    print(f"Part 1: {sum(freqs)}")
+    print(f"Part 2: {find_repeat(freqs)}")