7 files changed, 383 insertions, 6 deletions

diff --git a/aoc/__init__.py b/aoc/__init__.py
index b07ead1..70c3ac0 100644
--- a/aoc/__init__.py
+++ b/aoc/__init__.py
@@ -4,7 +4,7 @@ from collections import Iterator
 from typing import Generator, Any
 
 
-class AssignmentBase(ABC):
+class BaseAssignment(ABC):
     def __init__(self, path):
         self.path = path
 
diff --git a/aoc/__main__.py b/aoc/__main__.py
index da0496e..8c6a21e 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -26,4 +26,4 @@ if __name__ == '__main__':
     Assignment = getattr(assignment_day, f'Assignment{args.part}')
     assignment = Assignment(path=os.path.dirname(assignment_day.__file__))
 
-    print(assignment.run(input=assignment.read_input()))
+    print(assignment.run(input=assignment.read_input(example=args.example)))
diff --git a/day1/__init__.py b/day1/__init__.py
index ce728cc..cf48f78 100644
--- a/day1/__init__.py
+++ b/day1/__init__.py
@@ -1,9 +1,9 @@
 from typing import Iterator, List
 
-from aoc import AssignmentBase
+from aoc import BaseAssignment
 
 
-class Assignment(AssignmentBase):
+class Assignment(BaseAssignment):
     def parse_item(self, item: str) -> int:
         return int(item)
 
diff --git a/day2/__init__.py b/day2/__init__.py
index 988c32e..1d65d13 100644
--- a/day2/__init__.py
+++ b/day2/__init__.py
@@ -2,7 +2,7 @@ import re
 from dataclasses import dataclass
 from typing import Generator
 
-from aoc import AssignmentBase
+from aoc import BaseAssignment
 
 
 matcher = re.compile(
@@ -18,7 +18,7 @@ class Item:
     password: str
 
 
-class Assignment(AssignmentBase):
+class Assignment(BaseAssignment):
     def parse_item(self, item: str) -> Item:
         match = matcher.match(item).groupdict()
         return Item(
diff --git a/day3/__init__.py b/day3/__init__.py
new file mode 100644
index 0000000..4954c15
--- /dev/null
+++ b/day3/__init__.py
@@ -0,0 +1,43 @@
+from functools import reduce
+from typing import List, Generator
+
+from aoc import BaseAssignment
+
+
+class Assignment(BaseAssignment):
+    def read_input(self, example = False) -> List:
+        return list(super().read_input(example))
+
+    def slope(self, input: List, dx: int, dy: int) -> Generator:
+        for y, row in enumerate(input[::dy]):
+            x = (y * dx) % len(row)
+            yield (x, y)
+
+    def get_trees_for_slope(self, input: List, dx: int, dy: int):
+        trees = 0
+
+        for x, y in self.slope(input, dx, dy):
+            if x == 0 and y == 0:
+                continue
+
+            if (input[y * dy][x]) == '#':
+                trees += 1
+
+        return trees
+
+
+
+class AssignmentOne(Assignment):
+    def run(self, input: List):
+        return self.get_trees_for_slope(input, 3, 1)
+
+class AssignmentTwo(Assignment):
+    def run(self, input: List):
+        return reduce(lambda accumulator, value: accumulator * value, [
+            self.get_trees_for_slope(input, 1, 1),
+            self.get_trees_for_slope(input, 3, 1),
+            self.get_trees_for_slope(input, 5, 1),
+            self.get_trees_for_slope(input, 7, 1),
+            self.get_trees_for_slope(input, 1, 2),
+        ])
+
diff --git a/day3/example.txt b/day3/example.txt
new file mode 100644
index 0000000..8f551de
--- /dev/null
+++ b/day3/example.txt
@@ -0,0 +1,11 @@
+..##.......
+#...#...#..
+.#....#..#.
+..#.#...#.#
+.#...##..#.
+..#.##.....
+.#.#.#....#
+.#........#
+#.##...#...
+#...##....#
+.#..#...#.#
\ No newline at end of file
diff --git a/day3/input.txt b/day3/input.txt
new file mode 100644
index 0000000..6cfcb73
--- /dev/null
+++ b/day3/input.txt
@@ -0,0 +1,323 @@
+...#...#....#....##...###....#.
+#.#...#...#....#.........#..#..
+.#....##..#.#..##..##..........
+.....#.#.............#..#......
+.......#...#.##.#......#..#.#..
+#.#....#......##........##.....
+.....##.#....#..#...#...##...#.
+...#...#..#.......#..#...##...#
+..........#...........##.......
+..#..#..#...................#..
+#..#....#.....##.#.#..........#
+.#.##.......###.....#.#...#....
+.#..##....##....#.......#...###
+#.#..##...#.#..#...............
+.........#....#.......##.#.#...
+...###...##....##...#..##.#..#.
+....#.........#..#...#.......#.
+....................#..#.#.#...
+..#....#..........#...........#
+.#.....#..#.....##........##..#
+#..##..#...##............#..##.
+.#..##....#..........#..#.##.#.
+..#####..#.#............##.....
+...###.#....##..#.#....#.....#.
+.#.......##....#...#.#.....##..
+...#....#...##.#...#..#........
+.####.....#....#.#.#...#.......
+...#....#.....#.......#........
+#..#.#.......#...#............#
+...#.....###.##....#.#.###.#...
+.#.........#.......#.#....##...
+#.#..#...#.#...##......#..#....
+.....#...#..#.#...#..###..#....
+......#.........#...###........
+.....#..##...#..........#.....#
+..#..#.#.##.#...#....#....##..#
+##....#.##...#.##.#..##....#...
+.....#.#.#.#..#....##.#...#.#..
+.....##.......#........#.......
+...#.#.....#...#...##.#......##
+........#..#.#...#.#.....#.#..#
+#..##...#.#...##..##...#.#...##
+.##.#.#..#...#.....#.#.##.#...#
+.#.####.........##.........#..#
+.##..............#....#...#...#
+......#...#..#...#..#..###.#...
+.......##...#.#.#..##..#......#
+.....#....#..##..#.........#...
+.....#..#.#.#........#.#.####..
+#..#.......###....##...........
+#..##..........#.#......#.#....
+.....##........#...#..##.......
+###...#.##.#.#.#.#.##...##.....
+....#...#........##.#.##..##...
+.#..#.#.#......#.......##..#..#
+.#...#.................#....#..
+.##..#..........#..##.......#..
+.#.#.#.....#..#.#.........##..#
+...#......##...#.......#...##..
+##...###....#.###.............#
+#.....#.#..#.#..#........#.#.#.
+.....#.#......##..#.#.....#.##.
+.......#...........#..#.......#
+..#....#.#.#......#.....#...#..
+.....##........#..##..#..##....
+#.#........#...##....#.#..##...
+#......#......#....#..#...#.##.
+....#.#.......#.#.#............
+......####.#.##...#.#.##.....##
+..###.#.#..#.........#.####....
+.#.......#..#.#....#.#..#.#.##.
+#....#....#............##...##.
+....#....#............#....#..#
+..#........#..#....#..#..#...#.
+.#......##....#..........#....#
+#.##.....#..........#.###.#....
+....##...#.....#.#......#.##...
+#.#.....#.......###.###..#..#.#
+..###..##.............#.####.##
+#....#.....#....#..##.......#..
+.....#....#...#.#.#.#..#...#.##
+...#.....#..#....###......#.#.#
+##.........#.#..#..#.#..#.....#
+.#.....#.#....#.........##..#.#
+.#.#..#.###..#..#..........#...
+.##....#.#.#...#......##.....#.
+#.#....#....#...#...##...#..#..
+#...#........#....#....#......#
+#......#...#..#.#.##.....##..#.
+....#...#......##...#..#....#..
+.#......##.##.......#.......#..
+.#...#..####...........#.#.#...
+.........#...#.#.........#.....
+#.##.....#.#..#.#.###...###..#.
+#...##.###......#.###..##.#.##.
+...##.#.....#....#..#......#...
+#....###.#..#...##.....#......#
+........###...#...#............
+........#....#...#...#....#...#
+#....#..#..#....#.#........#.#.
+##...#.....#.#..........#..#..#
+#.#...##.....#........#...#...#
+##.#.#.......#...#..#.###....#.
+.#.......#....##..##...#.....#.
+#....#....#.....#.......#......
+.##.##.##...##...#.#.#..#..#...
+#..#..#.##....#......##....###.
+.......#.#.........#..##.#...##
+.#..##...#....#.....#..........
+..#.#...#......#.#..#..........
+.##....#.#.#.##.......###...#..
+..##.#...#.#.#.#.......#..#....
+#..#.......#...#........#.....#
+.....#.......#......###..#.....
+...##.#.......#.....##.....##..
+##..#.......#.#.....#....#.....
+..#....#.##.##...#...#......#..
+.#..#.###.#....###........#...#
+....##.##...##..#..#.#....#....
+..###...##.....##..............
+#....#...##...#....#..........#
+.##........#......##...##...#.#