7 files changed, 64 insertions, 42 deletions

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
new file mode 100644
index 0000000..1232470
--- /dev/null
+++ b/.pre-commit-config.yaml
@@ -0,0 +1,15 @@
+# See https://pre-commit.com for more information
+# See https://pre-commit.com/hooks.html for more hooks
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.3.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-merge-conflict
+      - id: debug-statements
+      - id: fix-encoding-pragma
+  - repo: https://github.com/psf/black
+    rev: 22.10.0
+    hooks:
+      - id: black
diff --git a/aoc/__init__.py b/aoc/__init__.py
index b13489c..5fce415 100644
--- a/aoc/__init__.py
+++ b/aoc/__init__.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import os
 from abc import ABC
 from typing import Generator, Any, Iterator
@@ -5,23 +6,24 @@ 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__}'
+        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'
+    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'
+            file = f"{self.path}/example.txt"
 
-        with open(file, 'r') as input_file:
+        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
+        raise NotImplementedError("Please implement run")
diff --git a/aoc/__main__.py b/aoc/__main__.py
index 25a7026..337bdfe 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import argparse
 import importlib
 import os
@@ -7,32 +8,37 @@ from typing import List
 
 def day(assignment_part: str) -> str:
     return {
-        '1': 'One',
-        '2': 'Two',
+        "1": "One",
+        "2": "Two",
     }[assignment_part]
 
+
 def kwargs(kwarg: str) -> List:
-    return kwarg.split('=')
+    return kwarg.split("=")
+
 
-parser = argparse.ArgumentParser(description='Advent of Code')
+parser = argparse.ArgumentParser(description="Advent of Code")
 
-parser.add_argument('day', type=str, help='Assignment day')
+parser.add_argument("day", type=str, help="Assignment day")
 parser.add_argument(
-    '-p', '--part', type=day, nargs='?', default='1',
-    help='Assingment part. Defaults to one.'
+    "-p",
+    "--part",
+    type=day,
+    nargs="?",
+    default="1",
+    help="Assingment part. Defaults to one.",
 )
-parser.add_argument('--example', default=False, action='store_true')
-parser.add_argument('kwargs', type=kwargs, nargs='*')
+parser.add_argument("--example", default=False, action="store_true")
+parser.add_argument("kwargs", type=kwargs, nargs="*")
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     args = parser.parse_args()
     assignment_day = importlib.import_module(args.day)
 
-    Assignment = getattr(assignment_day, f'Assignment{args.part}')
+    Assignment = getattr(assignment_day, f"Assignment{args.part}")
     assignment = Assignment(
-        path=os.path.dirname(assignment_day.__file__),
-        **dict(args.kwargs)
+        path=os.path.dirname(assignment_day.__file__), **dict(args.kwargs)
     )
 
     start = perf_counter()
diff --git a/aoc/test_init.py b/aoc/test_init.py
index a9155cc..f804c41 100644
--- a/aoc/test_init.py
+++ b/aoc/test_init.py
@@ -1,2 +1,6 @@
+# -*- coding: utf-8 -*-
 def test_assignment_examples(assignment):
-    assert assignment.run(input=assignment.read_input(example=True)) == assignment.example_result
+    assert (
+        assignment.run(input=assignment.read_input(example=True))
+        == assignment.example_result
+    )
diff --git a/aoc/utils.py b/aoc/utils.py
index 6d794f4..a5de050 100644
--- a/aoc/utils.py
+++ b/aoc/utils.py
@@ -1,5 +1,6 @@
+# -*- coding: utf-8 -*-
 from typing import Callable
 
 
 def bold(item: str, condition: Callable[[], bool]):
-    return f'\033[36m{item}\033[0m' if condition() else item
+    return f"\033[36m{item}\033[0m" if condition() else item
diff --git a/conftest.py b/conftest.py
index 24884e2..1c0a8e4 100644
--- a/conftest.py
+++ b/conftest.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import importlib
 import os
 from pkgutil import walk_packages
@@ -8,25 +9,28 @@ from aoc.__main__ import day
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
 
+
 def pytest_generate_tests(metafunc: Metafunc):
-    if 'assignment' in metafunc.fixturenames:
+    if "assignment" in metafunc.fixturenames:
         packages = [
             importlib.import_module(package.name)
             for package in walk_packages([dir_path])
         ]
 
         assignments = [
-            (getattr(package, f'Assignment{day(part)}', None), package)
+            (getattr(package, f"Assignment{day(part)}", None), package)
             for package in packages
-            for part in ['1', '2']
+            for part in ["1", "2"]
         ]
 
         metafunc.parametrize(
-            argnames=f'assignment',
+            argnames=f"assignment",
             argvalues=[
                 Assignment(path=package.__path__[0])
                 for (Assignment, package) in assignments
-                if Assignment is not None and hasattr(package, '__path__') and Assignment.example_result != NotImplemented
+                if Assignment is not None
+                and hasattr(package, "__path__")
+                and Assignment.example_result != NotImplemented
             ],
-            ids=lambda assignment: str(assignment)
+            ids=lambda assignment: str(assignment),
         )
diff --git a/day1/__init__.py b/day1/__init__.py
index d7c995c..e6e4ac9 100644
--- a/day1/__init__.py
+++ b/day1/__init__.py
@@ -1,19 +1,16 @@
+# -*- coding: utf-8 -*-
 from abc import ABC
 from itertools import groupby
 from typing import Iterator, Any, List
 
 from aoc import BaseAssignment
 
+
 class Assignment(BaseAssignment, ABC):
     def calculate_elf_calories(self, input: Iterator) -> List[int]:
         return [
-            sum([
-                int(i)
-                for i
-                in group
-            ])
-            for in_group, group
-            in groupby(input, key=bool)
+            sum([int(i) for i in group])
+            for in_group, group in groupby(input, key=bool)
             if in_group
         ]
 
@@ -29,11 +26,4 @@ class AssignmentTwo(Assignment):
     example_result = 45000
 
     def run(self, input: Iterator) -> int:
-        return sum(
+        return sum(sorted(self.calculate_elf_calories(input), reverse=True)[:3])