mirror of
https://github.com/qmk/qmk_cli.git
synced 2026-07-25 16:42:55 -04:00
Zach White
37 lines
928 B
Python
37 lines
928 B
Python
"""Useful helper functions.
|
|
"""
|
|
from milc import cli, format_ansi
|
|
|
|
|
|
def question(question, boolean=True, default=''):
|
|
"""Asks the user to answer a question.
|
|
|
|
This keeps re-asking until it gets acceptible input.
|
|
"""
|
|
if cli.args.yes:
|
|
return True
|
|
|
|
if default and default.lower() == 'y':
|
|
answer_key = 'Y/n'
|
|
elif default and default.lower() == 'n':
|
|
answer_key = 'y/N'
|
|
else:
|
|
answer_key = 'y/n'
|
|
|
|
prompt = format_ansi('\n*** %s [%s] ' % (question, answer_key))
|
|
|
|
while True:
|
|
answer = input(prompt)
|
|
print()
|
|
if answer == '' and default.lower() == 'y':
|
|
answer = 'y'
|
|
elif answer == '' and default.lower() == 'n':
|
|
answer = 'n'
|
|
|
|
if answer.lower() in ['y', 'yes']:
|
|
return True
|
|
elif answer.lower() in ['n', 'no']:
|
|
return False
|
|
else:
|
|
cli.echo('Invalid answer!')
|