04_password_processing.py

######################################
# --- Day 4: Passport Processing --- #
######################################

import AOCUtils

decChars = set("0123456789")
hexChars = set("0123456789abcdef")

checks1 = [
    lambda pp: all(field in pp for field in ["byr", "iyr", "eyr", "hgt", "hcl", "ecl", "pid"])
]

checks2 = [
    lambda pp: set(pp["byr"]) <= decChars and 1920 <= int(pp["byr"]) <= 2002,
    lambda pp: set(pp["iyr"]) <= decChars and 2010 <= int(pp["iyr"]) <= 2020,
    lambda pp: set(pp["eyr"]) <= decChars and 2020 <= int(pp["eyr"]) <= 2030,
    lambda pp: set(pp["hgt"][:-2]) <= decChars and \
               ((pp["hgt"][-2:] == "cm" and 150 <= int(pp["hgt"][:-2]) <= 193) or \
                (pp["hgt"][-2:] == "in" and 59 <= int(pp["hgt"][:-2]) <= 76)),
    lambda pp: len(pp["hcl"]) == 7 and pp["hcl"][0] == "#" and set(pp["hcl"][1:]) <= hexChars,
    lambda pp: pp["ecl"] in ["amb", "blu", "brn", "gry", "grn", "hzl", "oth"],
    lambda pp: len(pp["pid"]) == 9 and set(pp["pid"]) <= decChars
]

def isValid1(passport):
    return all(check(passport) for check in checks1)

def isValid2(passport):
    return isValid1(passport) and all(check(passport) for check in checks2)

######################################

rawInput = AOCUtils.loadInput(4)

for i in range(len(rawInput)):
    if rawInput[i] == "": rawInput[i] = "\n"

rawPassports = " ".join(rawInput).split(" \n ")

passports = []
for rawPassport in rawPassports:
    passport = dict()
    for kvp in rawPassport.split():
        k, v = kvp.split(":")
        passport[k] = v

    passports.append(passport)

p1 = sum(isValid1(passport) for passport in passports)
print("Part 1: {}".format(p1))

p2 = sum(isValid2(passport) for passport in passports)
print("Part 2: {}".format(p2))

AOCUtils.printTimeTaken()