def userquery(prompt, response=None, colors=None):
"""
Taken from Gentoo emerge tool and adapted to Mamba. This function
just ask a simple question to the user that shoudl be answered with
yes or no
:param prompt: the prompt to be shown to the user
:type prompt: str
:param response: a tuple or list with the posible responses
:type response: tuple or list
:param colors: a tuple ot list with the colors for responses
"""
if response is None:
response = ('Yes', 'No')
elif type(response) is not tuple and type(response) is not list:
print('Error: response must be tuple or list')
sys.exit(1)
if colors is None:
colors = (
create_color_func('darkgreen'),
create_color_func('darkred')
)
else:
if type(colors) is not tuple and type(colors) is not list:
print('Error: colors must be tuple or list')
sys.exit(1)
colors = tuple([create_color_func(color) for color in colors])
print('\n{}'.format(prompt), end=' ')
try:
while True:
result = raw_input('[{}] '.format(
'/'.join([colors[i](response[i])
for i in range(len(response))])
))
if result:
for key in response:
if result.upper() == key[:len(result)].upper():
return key
print(
'Sorry, response "{}" not understood.'.format(result),
end=' '
)
except (EOFError, KeyboardInterrupt):
print('Interrupted')
sys.exit(1)
except IndexError:
print('Response number is greater than colors tuple')
sys.exit(1)
def userchoice(prompt, choices, help):
"""
Choices menu for user input.
The user get a menu with help content for choices index followed by
a numeric index value. The user just choose a numeric index for his
choice and this choice is returned back
:param prompt: the prompt to be shown to the user
:type prompt: str
:param choices: the choices to be shown
:type choices: tuple or list
:param help: the help to be shown
:type help: tuple or list
"""
if help is None:
raise RuntimeError('Help can not be None')
if len(choices) != len(help):
raise RuntimeError(
'Choices and Help should has the same number of items'
)
if len(choices) == 0 or len(help) == 0:
raise RuntimeError('Nor choices or help can be empty')
color = create_color_func('blue')
print('\n{}'.format(prompt))
try:
while True:
for choice in choices:
print('{} [{}] '.format(
color(help[choices.index(choice)].ljust(30)),
bold(str(choices.index(choice)))
))
response = raw_input('[{}] '.format(
'/'.join([str(choices.index(choices[i])) for i in range(
len(choices))])
))
if response:
for key in choices:
# an empty response will match the first
# value in responses
if response == str(choices.index(key))[:len(response)]:
return key
print('Sorry, response "{}" not understood.'.format(response))
print('\n{}'.format(prompt))
except (EOFError, KeyboardInterrupt):
print('Interrupted')
sys.exit(1)
