from tools.aoc import AOCDay
from typing import Any


class Day(AOCDay):
    inputs = [
        [
            (522, "input1")
        ],
        [
            (73364, "input1")
        ]
    ]

    def part1(self) -> Any:
        freq = 0
        for f in self.getInputListAsType(int):
            freq += f

        return freq

    def part2(self) -> Any:
        freq = 0
        freq_set = set()
        while True:
            for f in self.getInputListAsType(int):
                freq_set.add(freq)
                freq += f
                if freq in freq_set:
                    return freq


if __name__ == '__main__':
    day = Day(2018, 1)
    day.run(verbose=True)