From d9296584270f1e902944fc7390d0b43c34aa2dbb Mon Sep 17 00:00:00 2001
From: nea <romangraef@gmail.com>
Date: Wed, 20 Apr 2022 17:49:59 +0200
Subject: literal tos violations

---
 sbdata/task.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'sbdata/task.py')

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
 
-- 
cgit