aoc-2020-python-day3-part-2.py

import string

def get_input():
  with open('input4full.txt') as f:
    passwords = f.read()
  return passwords.strip()


def run():
  passports = [p.replace('\n', ' ').split(' ') for p in get_input().split('\n\n')]

  tests = {
    'byr': lambda x: int(x) in range(1920, 2003),
    'iyr': lambda x: int(x) in range(2010, 2021),
    'eyr': lambda x: int(x) in range(2020, 2031),
    'hgt': lambda x: (x[-2:] in ['in', 'cm']) and {'in': lambda x: int(x) in range(59,77), 'cm': lambda x: int(x) in range(150,194)}[x[-2:]](x[:-2]),
    'hcl': lambda x: x[0] == '#' and all(c in string.hexdigits for c in x[1:]),
    'ecl': lambda x: x in {'amb', 'blu', 'brn', 'gry', 'grn', 'hzl', 'oth'},
    'pid': lambda x: all(y in string.digits for y in x) and len(x) == 9,
  }

  print(sum(all(tests[p[:3]](p[4:]) for p in passport if not p[:3] == 'cid') for passport in passports))

if __name__ == '__main__':
  run()