from aocd.models import Puzzle
from aocd import submit

class Elf:
    '''
        Elf object
    '''
    calories = list()

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

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

    def __lt__(self, other) -> bool:
        '''
            check if sum calories is less than other
        '''
        return (self.sum_calories() < other.sum_calories())

    def __repr__(self) -> str:
        return f'{str(self.calories)}'

def parse_input(data: list) -> list:
    '''
        parses the input data and generates a list of elfs
    '''
    # 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)
    data = puzzle.input_data
    elfs = parse_input(data)
    # sort elfs based on calories
    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 = 0
    for elf in elfs[:3]:
        answer += elf.sum_calories()
    print(f'answer part b: {answer}')
    submit(answer, part='b', day=1, year=2022)