public/py-tools
1
0
Fork 0
Code Issues Pull Requests Actions Releases 8 Activity

aoc.splitline(): allow return fields to have different types

Browse Source
This commit is contained in:
Stefan Harmuth 2021-12-02 06:20:34 +01:00
parent b8e54f51f5
commit ef0da77133
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aoc.py
View File

@ -96,7 +96,7 @@ class AOCDay:
return return_array
def getInputAsArraySplit(self, split_char: str = ',', return_type: Type = None) -> List:
def getInputAsArraySplit(self, split_char: str = ',', return_type: Union[Type, List[Type]] = None) -> List:
"""
get input for day x with the lines split by split_char
if input has only one line, returns a 1d array with the values
@ -139,5 +139,7 @@ def splitLine(line, split_char=',', return_type=None):
if return_type is None:
return line
elif isinstance(return_type, list):
return [return_type[x](i) if len(return_type) > x else i for x, i in enumerate(line)]
else:
return [return_type(i) for i in line]