From 2bfcdf09560081fc3a20abedc2e0bb0b05eae86e Mon Sep 17 00:00:00 2001
From: dece <shgck@pistache.land>
Date: Wed, 2 Dec 2020 14:03:05 +0100
Subject: [PATCH] Day 2

---
 2020/day2.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 2020/day2.py

diff --git a/2020/day2.py b/2020/day2.py
new file mode 100644
index 0000000..c559930
--- /dev/null
+++ b/2020/day2.py
@@ -0,0 +1,27 @@
+def main():
+    with open("day2.txt", "rt") as f:
+        lines = [line.rstrip() for line in f.readlines()]
+    # Part 1
+    valid = 0
+    for line in lines:
+        count, letter, password = line.split()
+        min_n, max_n = count.split("-")
+        letter = letter.rstrip(":")
+        if int(min_n) <= password.count(letter) <= int(max_n):
+            valid += 1
+    print("Valids:", valid)
+    # Part 2
+    valid = 0
+    for line in lines:
+        count, letter, password = line.split()
+        ofs_a, ofs_b = count.split("-")
+        letter = letter.rstrip(":")
+        a = password[int(ofs_a) - 1]
+        b = password[int(ofs_b) - 1]
+        if (a == letter or b == letter) and not (a == b == letter):
+            valid += 1
+    print("Valids:", valid)
+
+
+if __name__ == "__main__":
+    main()