summaryrefslogtreecommitdiff
path: root/lib/python/qmk/questions.py
diff options
context:
space:
mode:
authorskullydazed <skullydazed@users.noreply.github.com>2020-01-24 11:31:16 -0800
committerFlorian Didron <fdidron@users.noreply.github.com>2020-02-26 10:15:12 +0900
commit41b518c55fa1138e01fca28349cb131355ffd35f (patch)
treea79bb7a4f3aaaef4eca9695a87611be6d8e68a47 /lib/python/qmk/questions.py
parent72c5e4958cb136d1bda37c2369f71f8a1a6071ad (diff)
Beef up how `qmk doctor` works. (#7375)
* Beef up how `qmk doctor` works. * improve the `git submodule status` parsing. h/t @erovia * Fix whitespace and imports * yapf * Add documentation for the new doctor functionality * Replace type_unchanged() with str() * remove unused modules * Update lib/python/qmk/cli/doctor.py Co-Authored-By: Erovia <Erovia@users.noreply.github.com> Co-authored-by: Erovia <Erovia@users.noreply.github.com>
Diffstat (limited to 'lib/python/qmk/questions.py')
-rw-r--r--lib/python/qmk/questions.py97
1 files changed, 97 insertions, 0 deletions
diff --git a/lib/python/qmk/questions.py b/lib/python/qmk/questions.py
new file mode 100644
index 0000000000..34b0b43bc8
--- /dev/null
+++ b/lib/python/qmk/questions.py
@@ -0,0 +1,97 @@
+"""Functions to collect user input.
+"""
+
+from milc import cli, format_ansi
+
+
+def yesno(prompt, *args, default=None, **kwargs):
+ """Displays prompt to the user and gets a yes or no response.
+
+ Returns True for a yes and False for a no.
+
+ If you add `--yes` and `--no` arguments to your program the user can answer questions by passing command line flags.
+
+ @add_argument('-y', '--yes', action='store_true', arg_only=True, help='Answer yes to all questions.')
+ @add_argument('-n', '--no', action='store_true', arg_only=True, help='Answer no to all questions.')
+
+ Arguments:
+ prompt
+ The prompt to present to the user. Can include ANSI and format strings like milc's `cli.print()`.
+
+ default
+ Whether to default to a Yes or No when the user presses enter.
+
+ None- force the user to enter Y or N
+
+ True- Default to yes
+
+ False- Default to no
+ """
+ if not args and kwargs:
+ args = kwargs
+
+ if 'no' in cli.args and cli.args.no:
+ return False
+
+ if 'yes' in cli.args and cli.args.yes:
+ return True
+
+ if default is not None:
+ if default:
+ prompt = prompt + ' [Y/n] '
+ else:
+ prompt = prompt + ' [y/N] '
+
+ while True:
+ print()
+ answer = input(format_ansi(prompt % args))
+ print()
+
+ if not answer and prompt is not None:
+ return default
+
+ elif answer.lower() in ['y', 'yes']:
+ return True
+
+ elif answer.lower() in ['n', 'no']:
+ return False
+
+
+def question(prompt, *args, default=None, confirm=False, answer_type=str, **kwargs):
+ """Prompt the user to answer a question with a free-form input.
+
+ prompt
+ The prompt to present to the user. Can include ANSI and format strings like milc's `cli.print()`.
+
+ default
+ The value to return when the user doesn't enter any value. Use None to prompt until they enter a value.
+
+ answer_type
+ Specify a type function for the answer. Will re-prompt the user if the function raises any errors. Common choices here include int, float, and decimal.Decimal.
+ """
+ if not args and kwargs:
+ args = kwargs
+
+ if default is not None:
+ prompt = '%s [%s] ' % (prompt, default)
+
+ while True:
+ print()
+ answer = input(format_ansi(prompt % args))
+ print()
+
+ if answer:
+ if confirm:
+ if yesno('Is the answer "%s" correct?', answer, default=True):
+ try:
+ return answer_type(answer)
+ except Exception as e:
+ cli.log.error('Could not convert answer (%s) to type %s: %s', answer, answer_type.__name__, str(e))
+ else:
+ try:
+ return answer_type(answer)
+ except Exception as e:
+ cli.log.error('Could not convert answer (%s) to type %s: %s', answer, answer_type.__name__, str(e))
+
+ elif default is not None:
+ return default