126 lines
3.3 KiB
Python
126 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
class File:
|
||
|
|
def __init__(self, name, size):
|
||
|
|
self.name = name
|
||
|
|
self.size = size
|
||
|
|
|
||
|
|
|
||
|
|
class Directory:
|
||
|
|
def __init__(self, name, parent):
|
||
|
|
self.parent = parent
|
||
|
|
self.name = name
|
||
|
|
self.directories = []
|
||
|
|
self.files = []
|
||
|
|
|
||
|
|
def addFile(self, file):
|
||
|
|
self.files.append(file)
|
||
|
|
|
||
|
|
def addDirectory(self, directory):
|
||
|
|
self.directories.append(directory)
|
||
|
|
|
||
|
|
def getSize(self):
|
||
|
|
self.size = 0
|
||
|
|
for file in self.files:
|
||
|
|
self.size += int(file.size)
|
||
|
|
for directory in self.directories:
|
||
|
|
self.size += int(directory.getSize())
|
||
|
|
return self.size
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
|
||
|
|
class FileSystem:
|
||
|
|
def __init__(self):
|
||
|
|
self.root = Directory('/', '')
|
||
|
|
self.root.parent = self.root
|
||
|
|
self.current_directory = self.root
|
||
|
|
self.all_directories = []
|
||
|
|
|
||
|
|
def addFile(self, file):
|
||
|
|
self.root.files.append(file)
|
||
|
|
|
||
|
|
def addDirectory(self, directory):
|
||
|
|
self.root.directories.append(directory)
|
||
|
|
|
||
|
|
def getSize(self):
|
||
|
|
return self.root.getSize()
|
||
|
|
|
||
|
|
|
||
|
|
def change_directory(line, fs):
|
||
|
|
match line[2]:
|
||
|
|
case '..':
|
||
|
|
fs.current_directory = fs.current_directory.parent
|
||
|
|
case '/':
|
||
|
|
fs.current_directory = fs.root
|
||
|
|
case _:
|
||
|
|
for directory in fs.current_directory.directories:
|
||
|
|
if directory.name == line[2]:
|
||
|
|
fs.current_directory = directory
|
||
|
|
return
|
||
|
|
new_directory = Directory(line[2], fs.current_directory)
|
||
|
|
fs.all_directories.append(new_directory)
|
||
|
|
fs.current_directory.addDirectory(new_directory)
|
||
|
|
fs.current_directory = new_directory
|
||
|
|
|
||
|
|
|
||
|
|
def process_command(line, fs):
|
||
|
|
match line[1]:
|
||
|
|
case 'cd':
|
||
|
|
change_directory(line, fs)
|
||
|
|
case 'ls':
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def process_line(line, fs):
|
||
|
|
line = line.split()
|
||
|
|
match line[0]:
|
||
|
|
case '$':
|
||
|
|
process_command(line, fs)
|
||
|
|
case 'dir':
|
||
|
|
for directory in fs.current_directory.directories:
|
||
|
|
if directory.name == line[1]:
|
||
|
|
return
|
||
|
|
new_directory = Directory(line[1], fs.current_directory)
|
||
|
|
fs.all_directories.append(new_directory)
|
||
|
|
fs.current_directory.addDirectory(new_directory)
|
||
|
|
case _:
|
||
|
|
for file in fs.current_directory.files:
|
||
|
|
if file.name == line[1]:
|
||
|
|
file.size == line[0]
|
||
|
|
return
|
||
|
|
new_file = File(line[1], line[0])
|
||
|
|
fs.current_directory.addFile(new_file)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
with open('day7', 'r') as file:
|
||
|
|
lines = file.readlines()
|
||
|
|
|
||
|
|
fs = FileSystem()
|
||
|
|
fs.current_directory = fs
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
line = line.strip()
|
||
|
|
process_line(line, fs)
|
||
|
|
|
||
|
|
root_size = fs.getSize()
|
||
|
|
print(f'{fs.root.name} : {root_size}')
|
||
|
|
max_size = 30_000_000 - (70_000_000 - root_size)
|
||
|
|
smallest = 30_000_000
|
||
|
|
total_size = 0
|
||
|
|
for directory in fs.all_directories:
|
||
|
|
size = directory.getSize()
|
||
|
|
if size <= 100_000:
|
||
|
|
total_size += directory.getSize()
|
||
|
|
if size >= max_size:
|
||
|
|
if size < smallest:
|
||
|
|
smallest = size
|
||
|
|
print(total_size)
|
||
|
|
print(smallest)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|