Day2 part1

1 files changed, 44 insertions, 0 deletions

diff --git a/day2/__init__.py b/day2/__init__.py
new file mode 100644
index 0000000..e16787f
--- /dev/null
+++ b/day2/__init__.py
@@ -0,0 +1,44 @@
+from dataclasses import dataclass
+from typing import Iterator, List
+
+from aoc import BaseAssignment
+
+@dataclass
+class Instruction:
+    direction: str
+    steps: int
+
+class Assignment(BaseAssignment):
+    def parse_item(self, item: str) -> Instruction:
+        direction, steps = item.split(' ')
+        return Instruction(direction=direction, steps=int(steps))
+
+    def read_input(self, example = False) -> List[Instruction]:
+        return list(super().read_input(example))
+
+class AssignmentOne(Assignment):
+    depth = 0
+    horizontal = 0
+
+    def run(self, input: List[Instruction]) -> int:
+        for instruction in input:
+            match instruction.direction:
+                case 'forward':
+                    self.horizontal += instruction.steps
+                case 'up':
+                    self.depth -= instruction.steps
+                case 'down':
+                    self.depth += instruction.steps
+
+        return self.depth * self.horizontal
+
+
+
+class AssignmentTwo(Assignment):
+    def run(self, input: List) -> int:
+        new_input = [
+            input[i - 2] + input[i - 1] + input[i]
+            for i in range(2, len(input))
+        ]
+
+        return AssignmentOne(path='').run(new_input)