def query(prompt, default='', validators=None, batch=False):
# Set the nonempty validator as default
if validators is None:
validators = [RegexValidator(r'.+')]
# Let's build the prompt
if prompt[-1] is not ' ':
prompt += ' '
if default:
prompt += '[' + default + '] '
# If input is not valid keep asking
while True:
# If batch option is True then auto reply
# with default input
if not batch:
user_input = raw_input(prompt).strip() or default
else:
print(prompt)
user_input = ''
# Validate the user input
try:
for validator in validators:
user_input = validator(user_input)
return user_input
except Exception as e:
puts(yellow(e.message))
from clint.textui import colored, puts, prompt, validators as validators_module
from re import match
prompt_colors = {
"red": colored.red, "green": colored.green, "yellow": colored.yellow, "blue": colored.blue,
"black": colored.black, "magenta": colored.magenta, "cyan": colored.cyan, "white": colored.white
}
def new_query(prompt, color, default='', validators=None, batch=False):
# Set the nonempty validator as default
if validators is None:
validators = [validators_module.RegexValidator(r'.+')]
# Let's build the prompt
if prompt[-1] is not ' ':
prompt += ' '
if default:
prompt += '[' + default + '] '
# If input is not valid keep asking
while True:
# If batch option is True then auto reply
# with default input
if not batch:
# now the output is colored as expected
puts(prompt_colors[color](prompt))
user_input = input().strip() or default
else:
print(prompt)
user_input = ''
# Validate the user input
try:
for validator in validators:
user_input = validator(user_input)
return user_input
except Exception as e:
puts(yellow(e.message))
# now the query function is customized as per our needs
prompt.query = new_query
puts(colored.yellow("Hello!!!\n\n"))
user_input = prompt.query("Please enter something: ", "cyan")
以下是clint lib中prompt.py文件中查询函数的定义
您可以看到,您无法打印出您试图实现的彩色文本,因为它需要 一个文本字符串作为参数,我们可以改变查询函数,使其按预期工作,如下所示
注意:我认为这些变化足够清楚,不需要解释,但如果您有一些问题,我很乐意在评论部分回答
这不起作用的原因是prompt.py将您的输入转换为原始字符串,因此无法识别您选择的颜色。我认为这是一个糟糕的执行。你可以通过运行
来证明我的理论
user_number = prompt.query(str(colored.cyan('\nPlease enter something:\n')),batch=True)
这是一个无限循环,但它打印您的颜色只是因为它没有在这个case中转换为原始颜色
相关问题 更多 >
编程相关推荐