From 37aa8eec0498d7e8491084711132f16db9129a39 Mon Sep 17 00:00:00 2001
From: Tom van der Lee <tom@ScaryTerry.local>
Date: Tue, 15 Dec 2020 13:14:58 +0100
Subject: Added day 9

---
 aoc/__main__.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'aoc/__main__.py')

diff --git a/aoc/__main__.py b/aoc/__main__.py
index 8c6a21e..efb215d 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -2,6 +2,8 @@ import argparse
 import os
 import sys
 import importlib
+from typing import Tuple, List
+
 
 def day(assignment_part: str) -> str:
     return {
@@ -9,6 +11,9 @@ def day(assignment_part: str) -> str:
         '2': 'Two',
     }[assignment_part]
 
+def kwargs(kwarg: str) -> List:
+    return kwarg.split('=')
+
 parser = argparse.ArgumentParser(description='Advent of Code')
 
 parser.add_argument('day', type=str, help='Assignment day')
@@ -17,6 +22,7 @@ parser.add_argument(
     help='Assingment part. Defaults to one.'
 )
 parser.add_argument('--example', default=False, action='store_true')
+parser.add_argument('kwargs', type=kwargs, nargs='*')
 
 
 if __name__ == '__main__':
@@ -24,6 +30,13 @@ if __name__ == '__main__':
     assignment_day = importlib.import_module(args.day)
 
     Assignment = getattr(assignment_day, f'Assignment{args.part}')
-    assignment = Assignment(path=os.path.dirname(assignment_day.__file__))
+    assignment = Assignment(
+        path=os.path.dirname(assignment_day.__file__),
+        **dict(args.kwargs)
+    )
 
-    print(assignment.run(input=assignment.read_input(example=args.example)))
+    print(
+        assignment.run(
+            input=assignment.read_input(example=args.example),
+        )
+    )
-- 
cgit v1.2.3