updated day05

2020-12-05 06:26:33 +01:00
parent eb9c615c58
commit 21fd6de1cd
13 changed files with 3678 additions and 1 deletions
--- a/day04/module.py
+++ b/day04/module.py
@@ -0,0 +1,120 @@
+from aoc.input import get_input
+import re
+from aoc.partselector import part_one, part_two
+
+def pw(line):
+    return line.strip()
+
+inp = get_input(pw)
+
+fields = [
+    "byr",
+    "iyr",
+    "eyr",
+    "hgt",
+    "hcl",
+    "ecl",
+    "pid",
+]
+
+def p1():
+    inp = get_input(pw)
+    count = 0
+    passports = []
+    data = []
+    valid = 0
+    for i in inp:
+        if i == '':
+            d = ' '.join(data)
+            pd = d.split(' ')
+            lst = dict(map(lambda x: x.split(':'), pd))
+            v = True
+            for f in fields:
+                if f not in lst:
+                    v = False
+            if v :
+                valid += 1
+
+            print(v, sorted(lst.items(), key=lambda x:x[0]))
+            passports.append(' '.join(data))
+            data = []
+            continue
+        data.append(i)
+
+    passports.append(' '.join(data))
+    data = []
+    print(valid)
+
+fields2 = {"byr": "[0-9]{4}",
+           "iyr": "[0-9]{4}",
+           "eyr": "[0-9]{4}",
+           "hgt": "[0-9]*(in|cm)",
+           "hcl": "#[0-9a-f]{6}",
+           "ecl": "(amb|blu|brn|gry|grn|hzl|oth)",
+           "pid": "[0-9]{9}",
+          }
+
+def p2():
+    inp = get_input(pw)
+    count = 0
+    passports = []
+    data = []
+    valid = 0
+    for i in inp:
+        if i == '':
+            d = ' '.join(data)
+            pd = d.split(' ')
+            lst = dict(map(lambda x: x.split(':'), pd))
+            v = True
+            for f in fields2.keys():
+                print (f, end=" ")
+                if f not in lst:
+                    v = False
+                else:
+                    print("----->", fields2[f], lst[f])
+                if f in lst:
+                    v = v and re.match(fields2[f], lst[f]) is not None
+                    print (v, end=" ")
+                    if f == 'pid':
+                        v = v and len(lst[f]) == 9
+                        print (v, end=" ")
+                    if f == 'byr':
+                        v = v and int(lst[f]) >= 1920 and int(lst[f]) <= 2002
+                        print (v, end=" ")
+                    if f == 'iyr':
+                        v = v and int(lst[f]) >= 2010 and int(lst[f]) <= 2020
+                        print ( v, end=" ")
+                    if f == 'eyr':
+                        v = v and int(lst[f]) >= 2020 and int(lst[f]) <= 2030
+                        print ( v, end=" ")
+                    if f == 'hgt':
+                        m = re.match(fields2[f], lst[f])
+                        if m is not None:
+                            if m.group(1) == 'in':
+                                v = v and int(lst[f][:-2]) <= 76
+                                v = v and int(lst[f][:-2]) >= 59
+                            if m.group(1) == 'cm':
+                                v = v and int(lst[f][:-2]) >= 150
+                                v = v and int(lst[f][:-2]) <= 193
+                        print ( v, end=" ")
+                print()
+            if v :
+                valid += 1
+
+            if not v:
+                print()
+                print(v, sorted(lst.items(), key=lambda x:x[0]))
+                print()
+            passports.append(' '.join(data))
+            data = []
+            continue
+        data.append(i)
+    print(valid)
+    print('done')
+
+if part_one():
+    p1()
+
+
+if part_two():
+    p2()