from aocd.models import Puzzle
from aocd import submit


class Elf:
    """
    object which holds a a list of calories
    carried by this particular elf
    """

    calories = list()

    def __init__(self, numbers: list) -> None:
        """
        constructor
        """
        self.calories = numbers.copy()

    def __lt__(self, other) -> bool:
        """
        compare if sum of calorie is less than other
        """
        return self.sum_calories() < other.sum_calories()

    def __repr__(self) -> str:
        """
        print string representation of object
        """
        return f"{str(self.calories)}"

    def sum_calories(self) -> int:
        """
        calculate the total calories carried by this elf
        """
        return sum(self.calories)


def parse_input(data: list) -> list:
    """
    parses the input data and generates a list of elfs
    from the input data string
    """
    # split input string into list
    splits = data.split("\n")
    # split list into sublists
    numbers = list()
    tmp = list()
    for number in splits:
        if not number:
            numbers.append(tmp.copy())
            tmp.clear()
        else:
            tmp.append(int(number))
    # elf object for each sublist
    return [Elf(data) for data in numbers]


if __name__ == "__main__":
    # get puzzle and parse data
    puzzle = Puzzle(year=2022, day=1)

    # get a list of elfs and sort based on calories
    elfs = parse_input(puzzle.input_data)
    elfs = sorted(elfs, reverse=True)

    # part a: submit biggest number of calories
    answer = elfs[0].sum_calories()
    print(f"answer part a: {answer}")
    submit(answer, part="a", day=1, year=2022)

    # part b: submit the calorie sum of the top 3 elfs
    answer = sum([elf.sum_calories() for elf in elfs[:3]])
    print(f"answer part b: {answer}")
    submit(answer, part="b", day=1, year=2022)