aoc2021/01/b.py

import os, argparse, queue


def solve_task(filename):
    with open(filename) as infile:
        window = queue.Queue(3)
        current_sum = 0
        last_avg = None
        decrements = 0
        for line in infile:
            depth = int(line)
            
            if window.qsize() == 3:
                leaving_depth = window.get_nowait()
                current_sum -= leaving_depth
            
            current_sum += depth
            window.put_nowait(depth)
            
            if window.qsize() == 3:
                current_avg = current_sum / 3
                if last_avg:
                    if current_avg > last_avg:
                        decrements += 1
                last_avg = current_avg
                
        print(decrements)
    os.system("pause")

def parse_arguments():
    parser = argparse.ArgumentParser(description="Script that toggles output of a Tenma PSU",epilog="Have a nice day!")
    parser.add_argument('filename', nargs='?', default="infile.txt", help='Input file')
    args = parser.parse_args()
    return args

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

if __name__ == "__main__":
    main()