day7_simpler_notworking.py

import sys

# create a dictionary to store the size of each directory
sizes = {}

# initialize the current directory to the root
current_dir = '/'

# read the input file
with open(sys.argv[1], 'r') as input_file:
    # iterate over the lines in the file
    for line in input_file:
        # split the line on whitespace, excluding the $ symbol
        command, *args = line.strip().split(' ')[1:]
        # if the command is "cd", update the current directory
        if command == 'cd':
            current_dir = args[0]
        
        # if the command is "ls", update the sizes dictionary
        # with the sizes of the files in the current directory
        elif command == 'ls':
            for file in args:
                # split the file name and size
                name, size = file.split()
                # if the current directory is not in the dictionary,
                # add it and initialize its size to 0
                if current_dir not in sizes:
                    sizes[current_dir] = 0
                # add the size of the file to the size of the current directory
                sizes[current_dir] += int(size)

# print the total size of each directory
for directory, size in sizes.items():
    print(f'{directory}: {size}')