puzzle1.py (272B)
1 file = open('files/puzzle1.txt').readlines() 2 3 print(sum([int(int(mass) / 3) - 2 for mass in file])) 4 5 def calculate_fuel(mass): 6 fuel = int(mass / 3) - 2 7 return fuel + calculate_fuel(fuel) if fuel > 0 else 0 8 9 print(sum([calculate_fuel(int(mass)) for mass in file]))