d4-code.py

def readPass(inpath="day4-input.txt"):
    with open(inpath, "r") as infile:
        passports = infile.read().split('\n\n')
        return passports


def part1(passports):
    required = {'byr', 'iyr', 'eyr', 'hgt', 'hcl', 'ecl', 'pid'}
    count = 0
    for passport in passports:
        keys = set(map(lambda x: x.split(':')[0], passport.split()))
        if keys.issuperset(required):
            count += 1
    return count


def keyFilter(passports):
    required = {'byr', 'iyr', 'eyr', 'hgt', 'hcl', 'ecl', 'pid'}
    valid = []
    for passport in passports:
        passDict = {pair.split(':')[0]: pair.split(':')[1]
                    for pair in passport.split()}
        if set(passDict.keys()).issuperset(required):
            valid.append(passDict)
    return valid


def altPart1(passports):
    valid = keyFilter(passports)
    return len(valid)


def part2(passports):
    count = 0
    for passDict in keyFilter(passports):
        try:
            if (1920 <= int(passDict['byr']) <= 2002) \
                    and (2010 <= int(passDict['iyr']) <= 2020) \
                    and (2020 <= int(passDict['eyr']) <= 2030) \
                    and ((passDict['hgt'][-2:] == 'cm' and (150 <= int(passDict['hgt'][:-2]) <= 193))
                         or (passDict['hgt'][-2:] == 'in' and (59 <= int(passDict['hgt'][:-2]) <= 76))) \
                    and passDict['hcl'][0] == '#' and len(passDict['hcl']) == 7 and int(passDict['hcl'][1:], 16)+1 \
                    and passDict['ecl'] in {'amb', 'blu', 'brn', 'gry', 'grn', 'hzl', 'oth'} \
                    and (len(passDict['pid']) == 9 and passDict['pid'].isnumeric()):
                count += 1
        except:
            continue

    return count


def main():
    passports = readPass(inpath="day4-input.txt")
    print(f"Part 1: {altPart1(passports)}\nPart 2: {part2(passports)}")


main()