from typing import Any, Iterator

from ..base import BaseSolver


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 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))

        yield number2snafu(sum(map(snafu2number, lines)))