summertime warmup session with 2020 event

1 files changed, 42 insertions, 0 deletions

diff --git a/2020/aoc2020-d04.py b/2020/aoc2020-d04.py
new file mode 100644
index 0000000..22f044d
--- /dev/null
+++ b/2020/aoc2020-d04.py
@@ -0,0 +1,42 @@
+#advent of code 2020
+#day 04
+import re
+
+PuzzleInput = open("04.in","r").read()
+passports = PuzzleInput.split("\n\n");
+p1 = 0;
+p2 = 0;
+documents = dict(); 
+required = ["byr","iyr","eyr","hgt","hcl","ecl","pid"]; #puzzle description
+
+def part2(doc):
+	valid2 = True;
+	valid2 *= len(doc["byr"]) == 4 and int(doc["byr"]) >= 1920 and int(doc["byr"]) <= 2002;
+	valid2 *= len(doc["iyr"]) == 4 and int(doc["iyr"]) >= 2010 and int(doc["iyr"]) <= 2020; 
+	valid2 *= len(doc["eyr"]) == 4 and int(doc["eyr"]) >= 2020 and int(doc["eyr"]) <= 2030; 
+	valid2 *= doc["hgt"][-2:] == "cm" or doc["hgt"][-2:] == "in";
+	if doc["hgt"][-2:] == "cm": 
+		valid2 *= int(doc["hgt"][:-2]) >= 150 and int(doc["hgt"][:-2]) <= 193;
+	if doc["hgt"][-2:] == "in": 
+		valid2 *= int(doc["hgt"][:-2]) >= 59 and int(doc["hgt"][:-2]) <= 76;
+	valid2 *= doc["hcl"][0] == "#";
+	valid2 *= len(re.findall("[a-f]|[0-9]",doc["hcl"])) == 6;
+	valid2 *= doc["ecl"] in ["amb", "blu", "brn", "gry", "grn", "hzl", "oth"];
+	valid2 *= len(doc["pid"]) == 9;
+	return valid2; 
+
+for i,p in enumerate(passports):
+	p = p.replace("\n"," ")
+	documents[i] = dict();
+	for entry in p.split(" "):
+		if len(entry) == 0: continue;
+		k,v = entry.split(":");
+		documents[i][k] = v;	
+	valid = True;
+	for req in required:
+		valid *= req in documents[i];
+	p1 += 1*valid;
+	if valid: p2 += 1*part2(documents[i]);
+
+print("part 1 =",p1);
+print("part 2 =",p2);