Add day2 part1, WIP part2

day2/run.py

@@ -0,0 +1,53 @@
+from typing import List
+
+EXAMPLE = """
+7 6 4 2 1
+1 2 7 8 9
+9 7 6 2 1
+1 3 2 4 5
+8 6 4 4 1
+1 3 6 7 9
+"""
+
+
+class CheckLevels:
+    """Checks that levels are safe"""
+
+    def __init__(self, input: str):
+        self.levels: List[List[int]] = [
+            [int(val) for val in line.split()] for line in input.splitlines() if line
+        ]
+
+    def __str__(self):
+        return f"Part 1 = {self._part1}, Part 2 = {self._part2}"
+
+    @property
+    def _part1(self) -> int:
+        return len([val for val in self.get_safe(problem_damper=False) if val])
+
+    @property
+    def _part2(self) -> int:
+        return len([val for val in self.get_safe(problem_damper=True) if val])
+
+    def get_differences(self) -> List[List[int]]:
+        return [
+            [val - line[i] for i, val in enumerate(line[1:])] for line in self.levels
+        ]
+
+    def get_safe(self, problem_damper: bool = False) -> List[List[bool]]:
+        outp = []
+        for line in self.get_differences():
+            all_same = all([val < 0 for val in line]) or all([val > 0 for val in line])
+            all_ge_1 = min([abs(val) for val in line]) >= 1
+            all_le_3 = max([abs(val) for val in line]) <= 3
+            outp.append(all_same and all_ge_1 and all_le_3)
+        return outp
+
+
+if __name__ == "__main__":
+    example = CheckLevels(input=EXAMPLE)
+    print("EXAMPLE:\n", example)
+    assert example._part1 == 2
+    assert example._part2 == 4
+    with open("input.txt", "r") as f:
+        print("Puzzle:\n", CheckLevels(input=f.read()))