diff options
Diffstat (limited to 'sbdata/task.py')
-rw-r--r-- | sbdata/task.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/sbdata/task.py b/sbdata/task.py index 7011ca2..90f23e6 100644 --- a/sbdata/task.py +++ b/sbdata/task.py @@ -40,6 +40,9 @@ class Arguments: return question.ask() return value + def get_arg(self, label: str, arg_name: str, mapper: typing.Callable[[str], _T]) -> _T: + return mapper(self.get_value(label, self.args.get(arg_name), questionary.text('Missing ' + label))) + def has_flag(self, param: str) -> bool: return param in self.flags |