from tools.aoc import AOCDay


class Day(AOCDay):
    inputs = [
        [
            (2, "test_input01"),
            (2, "test_input01_2"),
            (654, "test_input01_3"),
            (33583, "test_input01_4"),
            (3282386, "input01")
        ],
        [
            (2, "test_input01_2"),
            (966, "test_input01_3"),
            (50346, "test_input01_4"),
            (4920708, "input01")
        ]
    ]

    def part1(self):
        fuel_sum = 0
        for x in self.getInputListAsType(int):
            fuel_sum += x // 3 - 2

        return fuel_sum

    def part2(self):
        fuel_sum = 0
        for x in self.getInputListAsType(int):
            fuel_add = x // 3 - 2
            while fuel_add > 0:
                fuel_sum += fuel_add
                fuel_add = fuel_add // 3 - 2

        return fuel_sum


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