solved day 1 challenge

1/elf_list.py

@@ -1,5 +1,67 @@
-from aocd import data
+from aocd.models import Puzzle
+from aocd import submit
 
-AOC_COOKIE = os.environ.get('AOC_SESSION_TOKEN', 'default')
+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)