Remove parse_input helper

This commit is contained in:
Anders Englöf Ytterström 2023-12-09 17:40:50 +01:00
parent 96c4c1383a
commit 08288964bd
11 changed files with 83 additions and 66 deletions

View file

@ -101,9 +101,8 @@ for i in [str(n).zfill(2) for n in range(1, 26)]:
["n", "title", "part_1", "part_2"],
0,
)
filepath = f"./input/{str(n).zfill(2)}.txt"
with open(filepath, "r") as f:
data = f.read()
with open(f"./input/{i}.txt", "r") as f:
data = f.read().strip()
headline(day.n, day.title)
try:
data = day.presolve(data)

View file

@ -1,5 +1,5 @@
import re
from output import answer, puzzleinput
from output import answer
n = 1
title = "Trebuchet?!"
@ -11,7 +11,7 @@ def part_1(data):
s = [int(c) for c in s if c.isdigit()]
return s[0] * 10 + s[-1]
return sum(value(line) for line in data)
return sum(value(line) for line in data.split())
@answer(2, "Calibration values sum: {}, including spelled out digits")
@ -37,19 +37,15 @@ def part_2(data):
]
return s[0] * 10 + s[-1]
return sum(value(line) for line in data)
@puzzleinput(n)
def parse_input(data):
return data.split()
return sum(value(line) for line in data.split())
if __name__ == "__main__":
parsed = parse_input()
with open(f"./input/01.txt", "r") as f:
inp = f.read().strip()
a = part_1(parsed)
b = part_2(parsed)
a = part_1(inp)
b = part_2(inp)
assert a == 54634
assert b == 53855

View file

@ -1,7 +1,7 @@
import re
from math import prod
from output import answer, puzzleinput
from collections import defaultdict
from output import answer
n = 2
title = "Cube Conundrum"
@ -38,16 +38,12 @@ def part_2(data):
return sum(power(game) for game in data.splitlines())
@puzzleinput(n)
def parse_input(data):
return data
if __name__ == "__main__":
parsed = parse_input()
with open(f"./input/02.txt", "r") as f:
inp = f.read().strip()
a = part_1(parsed)
b = part_2(parsed)
a = part_1(inp)
b = part_2(inp)
assert a == 2439
assert b == 63711

View file

@ -1,5 +1,5 @@
from collections import deque
from output import answer, puzzleinput
from output import answer
n = 3
title = "Gear Ratios"
@ -18,6 +18,7 @@ def part_2(presolved):
def presolve(data):
data = data.split()
adj = (-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)
w = len(data[0])
h = len(data)
@ -54,13 +55,12 @@ def presolve(data):
return sum(s), sum(gr)
@puzzleinput(n)
def parse_input(data):
return data.split()
if __name__ == "__main__":
parsed = presolve(parse_input())
with open(f"./input/03.txt", "r") as f:
inp = f.read().strip()
parsed = presolve(inp)
a = part_1(parsed)
b = part_2(parsed)

View file

@ -1,6 +1,6 @@
import re
from collections import defaultdict
from output import answer, puzzleinput
from output import answer
n = 4
title = "Scratchcards"
@ -18,11 +18,6 @@ def part_2(presolved):
return count
@puzzleinput(n)
def parse_input(data):
return data
def presolve(data):
scores = []
count = defaultdict(int)
@ -40,7 +35,9 @@ def presolve(data):
if __name__ == "__main__":
inp = parse_input()
with open(f"./input/04.txt", "r") as f:
inp = f.read().strip()
inp = presolve(inp)
a = part_1(inp)

View file

@ -2,7 +2,7 @@ import re
from itertools import repeat
from math import inf
from multiprocessing import Pool, freeze_support
from output import answer, puzzleinput
from output import answer
n = 5
title = "If You Give A Seed A Fertilizer"
@ -22,11 +22,6 @@ def part_2(data):
return _bruteforce(seeds, process, 8)
@puzzleinput(n)
def parse_input(data):
return data
def _bruteforce(seeds, process, p=1):
processes = [[tuple(map(int, line.split())) for line in step.splitlines()[1:]] for step in process]
@ -39,6 +34,7 @@ def _bruteforce(seeds, process, p=1):
sm += pool.starmap(_nearest, zip(parts, repeat(processes)))
return min(sm)
def _nearest(start_r, processes):
a, b = start_r
nearest = inf

View file

@ -1,5 +1,5 @@
from math import prod, sqrt, ceil, floor
from output import answer, puzzleinput
from output import answer
n = 6
title = "Wait For It"
@ -15,11 +15,6 @@ def part_2(presolved):
return presolved[1]
@puzzleinput(n)
def parse_input(data):
return data
def presolve(data):
values = data.split()
l = len(values) // 2
@ -40,7 +35,8 @@ def presolve(data):
if __name__ == "__main__":
inp = parse_input()
with open(f"./input/06.txt", "r") as f:
inp = f.read().strip()
inp = presolve(inp)

View file

@ -1,5 +1,5 @@
from collections import Counter
from output import answer, puzzleinput
from output import answer
n = 7
title = "Camel Cards"
@ -15,11 +15,6 @@ def part_2(data):
return _solve(data.replace("J", "0"))
@puzzleinput(n)
def parse_input(data):
return data
def _solve(data):
return sum(
c[-1] * i for i, c in enumerate(sorted(_hand(h) for h in data.splitlines()), 1)
@ -89,7 +84,8 @@ if __name__ == "__main__":
doctest.testmod()
inp = parse_input()
with open(f"./input/07.txt", "r") as f:
inp = f.read().strip()
a = part_1(inp)
b = part_2(inp)

View file

@ -1,6 +1,6 @@
import re
from math import lcm
from output import answer, puzzleinput
from output import answer
n = 8
title = "Haunted Wasteland"
@ -18,11 +18,6 @@ def part_2(presolved):
return ghost_meet_point
@puzzleinput(n)
def parse_input(data):
return data
def presolve(data):
d, els = data.split("\n\n")
r = len(d)
@ -54,7 +49,8 @@ def presolve(data):
if __name__ == "__main__":
inp = parse_input()
with open(f"./input/08.txt", "r") as f:
inp = f.read().strip()
inp = presolve(inp)

View file

@ -1,4 +1,4 @@
from output import answer, puzzleinput
from output import answer
n = 9
title = "Mirage Maintenance"
@ -30,7 +30,8 @@ def _solve(lines):
if __name__ == "__main__":
inp = parse_input()
with open(f"./input/09.txt", "r") as f:
inp = f.read().strip()
a = part_1(inp)
b = part_2(inp)

View file

@ -0,0 +1,44 @@
from output import answer
n = 10
title = "dddd"
@answer(1, "Answer is {}")
def part_1(data):
return data
@answer(2, "Actually, answer is {}")
def part_2(data):
return data
# uncomment to solve parts in one go
# def presolve(data):
# return data
if __name__ == "__main__":
# use dummy data
inp = """
replace me
""".strip()
# uncomment to instead use stdin
# import sys; inp = sys.stdin.read().strip()
# uncomment to use AoC provided puzzle input
# with open(f"./input/10.txt", "r") as f:
# inp = f.read()
# uncomment to do initial data processing shared by part 1-2
# inp = presolve(inp)
a = part_1(inp)
# b = part_2(inp)
# uncomment and replace 0 with actual output to refactor code
# and ensure nonbreaking changes
# assert a == 0
# assert b == 0