84 lines
2 KiB
Python
84 lines
2 KiB
Python
import re
|
|
|
|
|
|
def parse_input() -> list[dict]:
|
|
passports = [[]]
|
|
|
|
with open("input.txt") as file:
|
|
for line in file:
|
|
line_ = line.rstrip()
|
|
if len(line_) > 1:
|
|
passports[-1].append(line_)
|
|
else:
|
|
if not passports[-1] == []:
|
|
passports.append([])
|
|
|
|
parsed = []
|
|
|
|
for passport in passports:
|
|
tags = []
|
|
for line in passport:
|
|
tags += line.split()
|
|
pass_dict = {}
|
|
for tag in tags:
|
|
(name, value) = tag.split(":")
|
|
pass_dict[name] = value
|
|
parsed.append(pass_dict)
|
|
|
|
return parsed
|
|
|
|
|
|
PASSPORTS = parse_input()
|
|
|
|
|
|
def solve_p1() -> int:
|
|
|
|
def test_passport(passport: dict) -> bool:
|
|
for tag_name in ["byr", "iyr", "eyr", "hgt", "hcl", "ecl", "pid"]:
|
|
if tag_name not in passport:
|
|
return False
|
|
return True
|
|
|
|
counted = 0
|
|
for passport in PASSPORTS:
|
|
if test_passport(passport):
|
|
counted += 1
|
|
return counted
|
|
|
|
|
|
def solve_p2() -> int:
|
|
|
|
TAGS_REGEX = [
|
|
("byr", re.compile(r"((19[2-9]\d)|(200[012]))$")),
|
|
("iyr", re.compile(r"(201\d|2020)$")),
|
|
("eyr", re.compile(r"(202\d|2030)$")),
|
|
("hgt", re.compile(r"(((1[5-8]\d)|(19[0-3]))cm|((59|[67]\d)|7[0-6])in)$")),
|
|
("hcl", re.compile(r"#[0-9a-f]{6}$")),
|
|
("ecl", re.compile(r"(amb|blu|brn|gry|grn|hzl|oth)$")),
|
|
("pid", re.compile(r"\d{9}$"))
|
|
]
|
|
|
|
def test_tag(passport: dict, tag: str, pattern: re.Pattern) -> bool:
|
|
if tag not in passport:
|
|
return False
|
|
if pattern.match(passport[tag]):
|
|
return True
|
|
return False
|
|
|
|
def test_passport_adv(passport: dict) -> bool:
|
|
result = []
|
|
for (tag_name, pattern) in TAGS_REGEX:
|
|
result.append(test_tag(passport, tag_name, pattern))
|
|
return False not in result
|
|
|
|
counted = 0
|
|
for passport in PASSPORTS:
|
|
if test_passport_adv(passport):
|
|
counted += 1
|
|
|
|
return counted
|
|
|
|
|
|
print(solve_p1())
|
|
print(solve_p2())
|