Refactor code for API (#3)

Co-authored-by: Mikael CAPELLE <mikael.capelle@thalesaleniaspace.com>
Co-authored-by: Mikaël Capelle <capelle.mikael@gmail.com>
Reviewed-on: #3

src/holt59/aoc/2023/day8.py

@@ -1,29 +1,30 @@
 import itertools
 import math
-import sys
+from typing import Any, Iterator
 
-lines = sys.stdin.read().splitlines()
-
-sequence = lines[0]
-nodes = {
-    p[0]: {d: n for d, n in zip("LR", p[1].strip("()").split(", "))}
-    for line in lines[2:]
-    if (p := line.split(" = "))
-}
+from ..base import BaseSolver
 
 
-def path(start: str):
-    path = [start]
-    it_seq = iter(itertools.cycle(sequence))
-    while not path[-1].endswith("Z"):
-        path.append(nodes[path[-1]][next(it_seq)])
-    return path
+class Solver(BaseSolver):
+    def solve(self, input: str) -> Iterator[Any]:
+        lines = input.splitlines()
 
+        sequence = lines[0]
+        nodes = {
+            p[0]: {d: n for d, n in zip("LR", p[1].strip("()").split(", "))}
+            for line in lines[2:]
+            if (p := line.split(" = "))
+        }
 
-# part 1
-answer_1 = len(path(next(node for node in nodes if node.endswith("A")))) - 1
-print(f"answer 1 is {answer_1}")
+        def path(start: str):
+            path = [start]
+            it_seq = iter(itertools.cycle(sequence))
+            while not path[-1].endswith("Z"):
+                path.append(nodes[path[-1]][next(it_seq)])
+            return path
 
-# part 2
-answer_2 = math.lcm(*(len(path(node)) - 1 for node in nodes if node.endswith("A")))
-print(f"answer 2 is {answer_2}")
+        # part 1
+        yield len(path(next(node for node in nodes if node.endswith("A")))) - 1
+
+        # part 2
+        yield math.lcm(*(len(path(node)) - 1 for node in nodes if node.endswith("A")))