Initial code

1 files changed, 27 insertions, 0 deletions

diff --git a/aoc/__init__.py b/aoc/__init__.py
new file mode 100644
index 0000000..b13489c
--- /dev/null
+++ b/aoc/__init__.py
@@ -0,0 +1,27 @@
+import os
+from abc import ABC
+from typing import Generator, Any, Iterator
+
+
+class BaseAssignment(ABC):
+    example_result = NotImplemented
+    def __init__(self, path):
+        self.path = path
+
+    def __str__(self):
+        return f'{self.__module__}.{self.__class__.__name__}'
+
+    def parse_item(self, item: str) -> Any:
+        return item
+
+    def read_input(self, example = False) -> Generator:
+        file = f'{self.path}/input.txt'
+        if example or not os.path.isfile(file):
+            file = f'{self.path}/example.txt'
+
+        with open(file, 'r') as input_file:
+            for line in input_file.readlines():
+                yield self.parse_item(line.strip())
+
+    def run(self, input: Iterator) -> Any:
+        raise NotImplementedError('Please implement run')
\ No newline at end of file