64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
|
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()
|