This commit is contained in:
dece 2020-12-04 15:52:56 +01:00
parent 1f714736f5
commit e63eab257f

63
2020/day4.py Normal file
View file

@ -0,0 +1,63 @@
import re
import string
REQ = ["byr", "iyr", "eyr", "hgt", "hcl", "ecl", "pid"]
def main():
with open("day4.txt", "rt") as f:
lines = [line.rstrip() for line in f.readlines()]
passports = get_passports(lines)
# Part 1
num_valids = 0
for passport in passports:
if all(k in passport for k in REQ):
num_valids += 1
print("Valids:", num_valids)
# Part 2
num_valids = 0
hair_re = re.compile(r"#[abcdef\d]{6}")
for passport in passports:
if not all(k in passport for k in REQ):
continue
if not (1920 <= int(passport["byr"]) <= 2002):
continue
if not (2010 <= int(passport["iyr"]) <= 2020):
continue
if not (2020 <= int(passport["eyr"]) <= 2030):
continue
if not (
(passport["hgt"].endswith("cm") and 150 <= int(passport["hgt"][:-2]) <= 193)
or (passport["hgt"].endswith("in") and 59 <= int(passport["hgt"][:-2]) <= 76)
):
continue
if not hair_re.match(passport["hcl"]):
continue
if not passport["ecl"] in ["amb", "blu", "brn", "gry", "grn", "hzl", "oth"]:
continue
if not (len(passport["pid"]) == 9 and all(c in string.digits for c in passport["pid"])):
continue
num_valids += 1
print("P2 valids:", num_valids)
def get_passports(lines):
passports = []
passport = {}
for line in lines:
elements = line.split()
for element in elements:
k, v = element.split(":")
passport[k] = v
if not line:
passports.append(passport)
passport = {}
passports.append(passport)
return passports
if __name__ == "__main__":
main()