95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import getpass
|
|
from ._compat import string_types, input
|
|
|
|
|
|
def prompt(name, default=None):
|
|
"""
|
|
Grab user input from command line.
|
|
|
|
:param name: prompt text
|
|
:param default: default value if no input provided.
|
|
"""
|
|
|
|
prompt = name + (default and ' [%s]' % default or '')
|
|
prompt += name.endswith('?') and ' ' or ': '
|
|
while True:
|
|
rv = input(prompt)
|
|
if rv:
|
|
return rv
|
|
if default is not None:
|
|
return default
|
|
|
|
|
|
def prompt_pass(name, default=None):
|
|
"""
|
|
Grabs hidden (password) input from command line.
|
|
|
|
:param name: prompt text
|
|
:param default: default value if no input provided.
|
|
"""
|
|
|
|
prompt = name + (default and ' [%s]' % default or '')
|
|
prompt += name.endswith('?') and ' ' or ': '
|
|
while True:
|
|
rv = getpass.getpass(prompt)
|
|
if rv:
|
|
return rv
|
|
if default is not None:
|
|
return default
|
|
|
|
|
|
def prompt_bool(name, default=False, yes_choices=None, no_choices=None):
|
|
"""
|
|
Grabs user input from command line and converts to boolean
|
|
value.
|
|
|
|
:param name: prompt text
|
|
:param default: default value if no input provided.
|
|
:param yes_choices: default 'y', 'yes', '1', 'on', 'true', 't'
|
|
:param no_choices: default 'n', 'no', '0', 'off', 'false', 'f'
|
|
"""
|
|
|
|
yes_choices = yes_choices or ('y', 'yes', '1', 'on', 'true', 't')
|
|
no_choices = no_choices or ('n', 'no', '0', 'off', 'false', 'f')
|
|
|
|
while True:
|
|
rv = prompt(name, default and yes_choices[0] or no_choices[0])
|
|
if not rv:
|
|
return default
|
|
if rv.lower() in yes_choices:
|
|
return True
|
|
elif rv.lower() in no_choices:
|
|
return False
|
|
|
|
|
|
def prompt_choices(name, choices, default=None, no_choice=('none',)):
|
|
"""
|
|
Grabs user input from command line from set of provided choices.
|
|
|
|
:param name: prompt text
|
|
:param choices: list or tuple of available choices. Choices may be
|
|
single strings or (key, value) tuples.
|
|
:param default: default value if no input provided.
|
|
:param no_choice: acceptable list of strings for "null choice"
|
|
"""
|
|
|
|
_choices = []
|
|
options = []
|
|
|
|
for choice in choices:
|
|
if isinstance(choice, string_types):
|
|
options.append(choice)
|
|
else:
|
|
options.append("%s [%s]" % (choice[1], choice[0]))
|
|
choice = choice[0]
|
|
_choices.append(choice)
|
|
|
|
while True:
|
|
rv = prompt(name + ' - (%s)' % ', '.join(options), default).lower()
|
|
if rv in no_choice:
|
|
return None
|
|
if rv in _choices or rv == default:
|
|
return rv
|