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/2022/day10.py

@@ -1,38 +1,43 @@
-import sys
+from typing import Any, Iterator
 
-lines = sys.stdin.read().splitlines()
-
-cycle = 1
-x = 1
-
-values = {cycle: x}
-
-for line in lines:
-    cycle += 1
-
-    if line == "noop":
-        pass
-    else:
-        r = int(line.split()[1])
-
-        values[cycle] = x
-
-        cycle += 1
-        x += r
-
-    values[cycle] = x
-
-answer_1 = sum(c * values[c] for c in range(20, max(values.keys()) + 1, 40))
-print(f"answer 1 is {answer_1}")
+from ..base import BaseSolver
 
 
-for i in range(6):
-    for j in range(40):
-        v = values[1 + i * 40 + j]
+class Solver(BaseSolver):
+    def solve(self, input: str) -> Iterator[Any]:
+        lines = [line.strip() for line in input.splitlines()]
 
-        if j >= v - 1 and j <= v + 1:
-            print("#", end="")
-        else:
-            print(".", end="")
+        cycle, x = 1, 1
+        values = {cycle: x}
 
-    print()
+        for line in lines:
+            cycle += 1
+
+            if line == "noop":
+                pass
+            else:
+                r = int(line.split()[1])
+
+                values[cycle] = x
+
+                cycle += 1
+                x += r
+
+            values[cycle] = x
+
+        answer_1 = sum(c * values[c] for c in range(20, max(values.keys()) + 1, 40))
+        yield answer_1
+
+        yield (
+            "\n"
+            + "\n".join(
+                "".join(
+                    "#"
+                    if j >= (v := values[1 + i * 40 + j]) - 1 and j <= v + 1
+                    else "."
+                    for j in range(40)
+                )
+                for i in range(6)
+            )
+            + "\n"
+        )