From 89edb0a6d7fc302d5f04b973e976139163b08369 Mon Sep 17 00:00:00 2001 From: Stefan Harmuth Date: Fri, 1 Dec 2023 07:37:25 +0100 Subject: [PATCH] getInput variant, that only returns integers from input lines --- src/tools/aoc.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/tools/aoc.py b/src/tools/aoc.py index 06e0142..69e6f85 100644 --- a/src/tools/aoc.py +++ b/src/tools/aoc.py @@ -207,6 +207,12 @@ class AOCDay: else: return self.input.copy() + def getIntsFromInput(self) -> list: + if len(self.input) == 1: + return list(map(int, re.findall(r"-?\d+", self.input[0]))) + else: + return [list(map(int, re.findall(r"-?\d+", l))) for l in self.input] + def getMultiLineInputAsArray( self, return_type: Type = None, join_char: str = None ) -> List: