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/day25.py

@@ -1,27 +1,28 @@
-import sys
+from typing import Any, Iterator
 
-lines = sys.stdin.read().splitlines()
-
-coeffs = {"2": 2, "1": 1, "0": 0, "-": -1, "=": -2}
+from ..base import BaseSolver
 
 
-def snafu2number(number: str) -> int:
-    value = 0
-    for c in number:
-        value *= 5
-        value += coeffs[c]
-    return value
+class Solver(BaseSolver):
+    def solve(self, input: str) -> Iterator[Any]:
+        lines = [line.strip() for line in input.splitlines()]
 
+        coeffs = {"2": 2, "1": 1, "0": 0, "-": -1, "=": -2}
 
-def number2snafu(number: int) -> str:
-    values = ["0", "1", "2", "=", "-"]
-    res = ""
-    while number > 0:
-        mod = number % 5
-        res = res + values[mod]
-        number = number // 5 + int(mod >= 3)
-    return "".join(reversed(res))
+        def snafu2number(number: str) -> int:
+            value = 0
+            for c in number:
+                value *= 5
+                value += coeffs[c]
+            return value
 
+        def number2snafu(number: int) -> str:
+            values = ["0", "1", "2", "=", "-"]
+            res = ""
+            while number > 0:
+                mod = number % 5
+                res = res + values[mod]
+                number = number // 5 + int(mod >= 3)
+            return "".join(reversed(res))
 
-answer_1 = number2snafu(sum(map(snafu2number, lines)))
-print(f"answer 1 is {answer_1}")
+        yield number2snafu(sum(map(snafu2number, lines)))