2022 Day 7 Python rough

2022/day07.py

@@ -0,0 +1,54 @@
+with open('input/7', 'r') as file:
+	lines = file.read().strip().split('\n')
+pwd = []
+files = {}
+
+for line in lines:
+	# print(pwd, line)
+	if line[0] == '$':
+		if line[2:4] == 'cd':
+			_, _, newdir = line.rpartition(' ')
+			if newdir == '/':
+				pwd = []
+			elif newdir.startswith('/'):
+				pwd = newdir[1:].split('/')
+			elif newdir == '..':
+				pwd.pop()
+			else:
+				pwd += newdir.split('/')
+		elif line[2:4] == 'ls':
+			pass
+	elif line[:3] == 'dir':
+		pass
+	else:
+		size, _, filename = line.partition(' ')
+		l = int(size)
+		dir = files
+		for d in pwd:
+			dir[d] = dir.get(d, {})
+			dir = dir[d]
+		dir[filename] = l
+
+dir_sizes = {}
+def get_dir_size(dir_name, dir):
+	if dir_name in dir_sizes:
+		return dir_sizes[dir_name]
+	s = 0
+	for filename, v in dir.items():
+		if isinstance(v, dict):
+			s += get_dir_size(dir_name + '/' + filename, v)
+		else:
+			s += v
+	dir_sizes[dir_name] = s
+	return s
+get_dir_size('/', files)
+
+print(f'Part 1: {sum([v for v in dir_sizes.values() if v <= 100000])}')  # 1118405
+
+
+total_capacity = 70_000_000
+need = 30_000_000
+remaining_capacity = total_capacity - dir_sizes['/']
+target = need - remaining_capacity
+
+print(f'Part 2: {sorted([v for v in dir_sizes.values() if v >= target])[0]}')