2020 day 04 (not cute)

2020/day04.py

@@ -0,0 +1,39 @@
+with open('input04', 'r') as file:
+  lines = [l.replace('\n', ' ') for l in file.read().split('\n\n')]
+passports = [{k:v for k,_,v in [token.partition(':') for token in line.split()]} for line in lines]
+
+req_keys = {'byr', 'iyr', 'eyr', 'hgt', 'hcl', 'ecl', 'pid'}
+valid_passports = [p for p in passports if req_keys.issubset(p.keys())]
+print(f'Part 1: {len(valid_passports)} valid passports')
+
+# Absolutely nothing cute about this validation function
+valid_ecl = {'amb', 'blu', 'brn', 'gry', 'grn', 'hzl', 'oth'}
+valid_hgt = {'cm': lambda i: (150<=i<=193), 'in': lambda i: (59<=i<=76)}
+def validate_pp(pp):
+  try:
+    if not (1920 <= int(pp['byr']) <= 2002):
+      return False
+    if not (2010 <= int(pp['iyr']) <= 2020):
+      return False
+    if not (2020 <= int(pp['eyr']) <= 2030):
+      return False
+    if not valid_hgt[pp['hgt'][-2:]](int(pp['hgt'][:-2])):
+      return False
+    hcl = pp['hcl']
+    if hcl[0] != '#':
+      return False
+    for c in hcl[1:]:
+      if c not in {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'}:
+        return False
+    if pp['ecl'] not in valid_ecl:
+      return False
+    if len(pp['pid'])!=9:
+      return False
+    int(pp['pid'])
+    return True
+  except:
+    #raise
+    return False
+
+
+print(f'Part 2: {[validate_pp(pp) for pp in valid_passports].count(True)} valid passports')