From dd2fc5c095afc1a0a30a8a8ba4678f27928d82fb Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Tue, 6 Dec 2022 10:30:30 +0100
Subject: [PATCH] [py] Match rust input and output format

---
 py/aoc/__init__.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/py/aoc/__init__.py b/py/aoc/__init__.py
index e66b473..b752d90 100644
--- a/py/aoc/__init__.py
+++ b/py/aoc/__init__.py
@@ -17,19 +17,23 @@ DAYS = {
 }
 
 
+def eprint(*args, **kwargs):
+    print(*args, **kwargs, file=sys.stderr)
+
+
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("files", type=Path, nargs="+")
     args = parser.parse_args()
 
     for file in args.files:
-        day = DAYS.get(file.stem)
+        day = DAYS.get(file.stem[:7])
         if day is None:
-            print(f"### Can't solve {file}", file=sys.stderr)
+            eprint(f"### Can't solve {file}")
             continue
 
-        print(f"### Solving day {file.stem}")
+        eprint(f"### Solving {file}")
         with open(file) as f:
             inputstr = f.read()
         day(inputstr)
-        print()
+        eprint()