如何重复try-except块

时间:2013-03-05 19:28:46

标签: python exception-handling python-3.x

我在Python 3.3中有一个try-except块,我想让它无限期地运行。

try:
    imp = int(input("Importance:\n\t1: High\n\t2: Normal\n\t3: Low"))
except ValueError:
    imp = int(input("Please enter a number between 1 and 3:\n> ")

目前,如果用户要输入非整数,它将按计划运行,但是如果他们再次输入,则只会再次引发ValueError并崩溃。

解决此问题的最佳方法是什么?

2 个答案:

答案 0 :(得分:22)

将它放在while循环中,当你得到预期的输入时就会爆发。最好让所有代码都依赖于imp中的try,如下所示,或为其设置默认值以防止NameError进一步下降。

while True:
  try:
    imp = int(input("Importance:\n\t1: High\n\t2: Normal\n\t3: Low"))

    # ... Do stuff dependant on "imp"

    break # Only triggered if input is valid...
  except ValueError:
    print("Error: Invalid number")

编辑:user2678074提出了一个有效点,这可能会使调试变得困难,因为它可能会陷入无限循环。

我会提出两个建议来解决这个问题 - 首先使用带有定义重试次数的for循环。其次,将上面的内容放在一个函数中,因此它与应用程序逻辑的其余部分保持独立,并且错误在该函数的范围内被隔离:

def safeIntegerInput( num_retries = 3 ):
    for attempt_no in range(num_retries):
        try:
            return int(input("Importance:\n\t1: High\n\t2: Normal\n\t3: Low"))
        except ValueError as error:
            if attempt_no < (num_retries - 1):
                print("Error: Invalid number")
            else:
                raise error

有了这个,你可以尝试/除了函数调用之外,只有当你超过最大重试次数时才会这样做。

答案 1 :(得分:7)

prompt = "Importance:\n\t1: High\n\t2: Normal\n\t3: Low\n> "
while True:
    try:
        imp = int(input(prompt))
        if imp < 1 or imp > 3:
            raise ValueError
        break
    except ValueError:
        prompt = "Please enter a number between 1 and 3:\n> "

输出:

rob@rivertam:~$ python3 test.py 
Importance:
    1: High
    2: Normal
    3: Low
> 67
Please enter a number between 1 and 3:
> test
Please enter a number between 1 and 3:
> 1
rob@rivertam:~$