如果用户输入无效答案，如何重做输入

Question

我是编程新手，我想知道如果用户输入无效数据，我该如何重复输入部分。

我希望应用程序只重复输入部分，而不是重新运行函数并让用户重新输入所有内容。

我的猜测是我必须将“return main()”更改为其他内容。

condition = input("What is the condition of the phone(New or Used)?")
if condition not in ["New", "new", "Used", "used"]:
    print("Invalid input")
    return main()

gps = input("Does the phone have gps(Yes or No)?")
if gps not in ["Yes", "yes", "No", "no"]:
    print("Invalid input")
    return main()

Answer 1

您可以创建一个方法来循环检查它：

def check_input(values,  message):
    while True:
        x = input(message) 
        if x in values:
            return x
        print "invalid values, options are "   + str(values) 

Answer 2

您可以概括代码以使用消息提示和验证功能：

def validated_input(prompt, validate):
    valid_input = False
    while not valid_input:
        value = input(prompt)
        valid_input = validate(value)
    return value

例如：

>>> def new_or_used(value):
...     return value.lower() in {"new", "used"}

>>> validate_input("New, or used?", new_or_used)

或者，更简单，但不太灵活，传入有效值：

def validated_input(prompt, valid_values):
    valid_input = False
    while not valid_input:
        value = input(prompt)
        valid_input = value.lower() in valid_values
    return value

并使用：

>>> validate_input("New, or used?", {"new", "used"})

您甚至可以使用有效值来创建输入提示：

def validated_input(prompt, valid_values):
    valid_input = False
    while not valid_input:
        value = input(prompt + ': ' + '/'.join(valid_values))
        valid_input = value.lower() in valid_values
    return value

提示：

>>> validate_input("What is the condition of the phone?", {"new", "used"})
What is the condition of the phone?: new/used

Answer 3

这是关于Control Flows 的好读物。

在您的情况下，您也可以使用strip() 和lower() 进行用户输入。

>>> 'HeLLo'.lower()
'hello'
>>> ' hello   '.strip()
'hello'

这里是 Python 3 的解决方案：

while True:
    condition=input("What is the condition of the phone(New or Used)?")
    if condition.strip().lower() in ['new', 'used']:
        break
    print("Invalid input")

while True:
    gps=input("Does the phone have gps(Yes or No)?")
    if gps.strip().lower() in ['yes','no']:
        break
    print("Invalid input")