seba
/
clintermission
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Check type of options passed to __init__

This commit is contained in:
Sebastian Lohff 2020-08-05 03:09:24 +02:00
parent c74e3d68d6
commit 9b833db364
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clintermission/climenu.py
View File

@ -111,10 +111,15 @@ class CliMenu:
if options: if options:
for option in options: for option in options:
if isinstance(option, tuple) and len(option) == 2: if isinstance(option, tuple):
self.add_option(*option) self.add_option(*option)
else: elif isinstance(option, dict):
self.add_option(**option)
elif isinstance(option, str):
self.add_option(option, option) self.add_option(option, option)
else:
raise ValueError("Option needs to be either tuple, dict or string, found '{}' of type {}"
.format(option, type(option)))
def add_header(self, *args, **kwargs): def add_header(self, *args, **kwargs):
return self.add_text(*args, **kwargs) return self.add_text(*args, **kwargs)