98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import getpass
|
||
|
|
from ._compat import string_types, ascii_lowercase, 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, resolve=ascii_lowercase,
|
||
|
|
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)
|
||
|
|
if not rv:
|
||
|
|
return default
|
||
|
|
rv = resolve(rv)
|
||
|
|
if rv in no_choice:
|
||
|
|
return None
|
||
|
|
if rv in _choices:
|
||
|
|
return rv
|