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

@@ -1,7 +1,8 @@
 import copy
-import sys
 from functools import reduce
-from typing import Callable, Final, Mapping, Sequence
+from typing import Any, Callable, Final, Iterator, Mapping, Sequence
+
+from ..base import BaseSolver
 
 
 class Monkey:
@@ -119,24 +120,28 @@ def monkey_business(inspects: dict[Monkey, int]) -> int:
     return sorted_levels[-2] * sorted_levels[-1]
 
 
-monkeys = [parse_monkey(block.splitlines()) for block in sys.stdin.read().split("\n\n")]
+class Solver(BaseSolver):
+    def solve(self, input: str) -> Iterator[Any]:
+        monkeys = [parse_monkey(block.splitlines()) for block in input.split("\n\n")]
 
-# case 1: we simply divide the worry by 3 after applying the monkey worry operation
-answer_1 = monkey_business(
-    run(copy.deepcopy(monkeys), 20, me_worry_fn=lambda w: w // 3)
-)
-print(f"answer 1 is {answer_1}")
+        # case 1: we simply divide the worry by 3 after applying the monkey worry operation
+        yield monkey_business(
+            run(copy.deepcopy(monkeys), 20, me_worry_fn=lambda w: w // 3)
+        )
 
-# case 2: to keep reasonable level values, we can use a modulo operation, we need to
-# use the product of all "divisible by" test so that the test remains valid
-#
-# (a + b) % c == ((a % c) + (b % c)) % c --- this would work for a single test value
-#
-# (a + b) % c == ((a % d) + (b % d)) % c --- if d is a multiple of c, which is why here
-# we use the product of all test value
-#
-total_test_value = reduce(lambda w, m: w * m.test_value, monkeys, 1)
-answer_2 = monkey_business(
-    run(copy.deepcopy(monkeys), 10_000, me_worry_fn=lambda w: w % total_test_value)
-)
-print(f"answer 2 is {answer_2}")
+        # case 2: to keep reasonable level values, we can use a modulo operation, we need to
+        # use the product of all "divisible by" test so that the test remains valid
+        #
+        # (a + b) % c == ((a % c) + (b % c)) % c --- this would work for a single test value
+        #
+        # (a + b) % c == ((a % d) + (b % d)) % c --- if d is a multiple of c, which is why here
+        # we use the product of all test value
+        #
+        total_test_value = reduce(lambda w, m: w * m.test_value, monkeys, 1)
+        yield monkey_business(
+            run(
+                copy.deepcopy(monkeys),
+                10_000,
+                me_worry_fn=lambda w: w % total_test_value,
+            )
+        )