AdventOfCode/2022/01/b.py

import os, argparse


def solve_task(filename):
    with open(filename) as infile:
        all_elves = []
        current_elf = 0
        highest = 0
        for raw_line in infile:
            line = raw_line.rstrip()
            if line != "":
                current_elf += int(line)
            else:
                all_elves.append(current_elf)
                highest = max(current_elf, highest)
                current_elf = 0
        
        all_elves.append(current_elf)
        highest = max(current_elf, highest)
        current_elf = 0

        all_elves.sort(reverse=True)

        to_sum = all_elves[0:3]
        three_highest = sum(to_sum)
        print(f"Three highest ammount {three_highest}")
            

def parse_arguments():
    parser = argparse.ArgumentParser(description="Script that solves the case",epilog="Have a nice day!")
    parser.add_argument('filename', nargs='?', default="example.txt", help='Input file')
    args = parser.parse_args()
    return args

def main():
    args = parse_arguments()
    solve_task(args.filename)

if __name__ == "__main__":
    main()