From 01e2920a67beff11d3bca01b2be595b7a6863eaa Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Tue, 6 Dec 2022 20:05:38 +0100
Subject: [PATCH] [py] Port 2015_01

---
 py/2015/01/solve.py | 36 ------------------------------------
 py/aoc/__init__.py  |  2 ++
 py/aoc/y2015/d01.py | 29 +++++++++++++++++++++++++++++
 3 files changed, 31 insertions(+), 36 deletions(-)
 delete mode 100644 py/2015/01/solve.py
 create mode 100644 py/aoc/y2015/d01.py

diff --git a/py/2015/01/solve.py b/py/2015/01/solve.py
deleted file mode 100644
index f295fd9..0000000
--- a/py/2015/01/solve.py
+++ /dev/null
@@ -1,36 +0,0 @@
-import sys
-
-def load_steps(filename):
-	with open(filename, "r") as f:
-		return f.read()[:-1]
-
-# PART 1
-
-def count_floors(steps):
-	return steps.count("(") - steps.count(")")
-
-# PART 2
-
-def find_basement_char(steps):
-	pos = 0
-	step_nr = 0
-	for step in steps:
-		step_nr += 1
-		if step == "(":
-			pos += 1
-		if step == ")":
-			pos -= 1
-		if pos < 0:
-			return step_nr
-
-def main(filename):
-	steps = load_steps(filename)
-	print(f"Solutions for {filename}")
-	floor = count_floors(steps)
-	print(f"Part 1: {floor}")
-	step_nr = find_basement_char(steps)
-	print(f"Part 2: {step_nr}")
-
-if __name__ == "__main__":
-	for filename in sys.argv[1:]:
-		main(filename)
diff --git a/py/aoc/__init__.py b/py/aoc/__init__.py
index de29880..519a2d5 100644
--- a/py/aoc/__init__.py
+++ b/py/aoc/__init__.py
@@ -2,6 +2,7 @@ import sys
 import argparse
 from pathlib import Path
 
+from .y2015 import d01
 from .y2016 import d01, d02, d03, d04, d05
 from .y2017 import d01, d02, d03, d04, d05, d06, d07, d08, d09
 from .y2018 import d01, d02, d03, d04, d05, d06, d07, d08, d09, d10, d11
@@ -10,6 +11,7 @@ from .y2021 import d14
 from .y2022 import d01, d02, d03, d04, d05, d06
 
 DAYS = {
+    "2015_01": y2015.d01.solve,
     "2016_01": y2016.d01.solve,
     "2016_02": y2016.d02.solve,
     "2016_03": y2016.d03.solve,
diff --git a/py/aoc/y2015/d01.py b/py/aoc/y2015/d01.py
new file mode 100644
index 0000000..3aee110
--- /dev/null
+++ b/py/aoc/y2015/d01.py
@@ -0,0 +1,29 @@
+# PART 1
+
+
+def count_floors(steps):
+    return steps.count("(") - steps.count(")")
+
+
+# PART 2
+
+
+def find_basement_char(steps):
+    pos = 0
+    step_nr = 0
+    for step in steps:
+        step_nr += 1
+        if step == "(":
+            pos += 1
+        if step == ")":
+            pos -= 1
+        if pos < 0:
+            return step_nr
+
+
+def solve(inputstr):
+    steps = inputstr.strip()
+    floor = count_floors(steps)
+    print(f"Part 1: {floor}")
+    step_nr = find_basement_char(steps)
+    print(f"Part 2: {step_nr}")