aboutsummaryrefslogtreecommitdiff
path: root/sbdata/task.py
diff options
context:
space:
mode:
Diffstat (limited to 'sbdata/task.py')
-rw-r--r--sbdata/task.py3
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