from aoc import AOC def get_solution(inp: str): mass_data = [int(item) for item in inp.split('\n')] fuel_data = [] for mass in mass_data: # fuel required for the module fuel = get_fuel(mass) # fuel required for the fuel itself while fuel > 0: fuel_data.append(fuel) fuel = get_fuel(fuel) total_fuel = sum(fuel_data) return total_fuel def get_fuel(mass: int): return mass // 3 - 2 aoc_data = AOC.get_data(1) result = get_solution(aoc_data) print(result) # AOC.submit(data=result, part='b', day=1)
from aoc import AOC from year_20.day_13.solution_b import get_solution day = 13 part = "b" aoc_data = AOC.get_data(day=day, year=2020) result = get_solution(aoc_data) print(result) AOC.submit(data=result, part=part, day=day, year=2020)