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

make AOCDay.getInput() return type casted input

Browse Source
This commit is contained in:
Stefan Harmuth 2022-12-03 11:10:03 +01:00
parent a2aa81388c
commit 353ff51411
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tools/aoc.py
View File

@ -168,17 +168,17 @@ class AOCDay:
answer_cache.save()
def getInput(self) -> Union[str, List]:
def getInput(self, return_type: Type = None) -> Any:
if len(self.input) == 1:
return self.input[0]
if return_type:
return return_type(self.input[0])
else:
return self.input[0]
else:
return self.input.copy()
def getInputListAsType(self, return_type: Type) -> List:
"""
get input as list casted to return_type, each line representing one list entry
"""
return [return_type(i) for i in self.input]
if return_type:
return [return_type(i) for i in self.input]
else:
return self.input.copy()
def getMultiLineInputAsArray(self, return_type: Type = None, join_char: str = None) -> List:
"""