输入异常处理，让 int() 更加优雅安全

本文将指导你如何在 Python 中优雅地处理 int() 函数转换用户输入时可能出现的异常，特别是 UnboundLocalError。

理解问题

在尝试直接将用户输入转换为整数时，如果用户输入了非数字字符，int() 函数会抛出 ValueError 异常。然而，如果在 try 块中发生异常，并且在 except 块中尝试引用在 try 块中定义的变量，但该变量由于异常而未被赋值，则会引发 UnboundLocalError。

例如，以下代码可能会导致问题：

def user_input():
    try:
        a = int(input("Enter a number: "))
    except ValueError:
        print(f"\n{a} is invalid.\n")
    else:
        return a

如果用户输入了非数字字符，int(input(...)) 会抛出 ValueError，导致 a 没有被赋值。然后，except 块中的 print(f"\n{a} is invalid.\n") 尝试引用未赋值的 a，从而引发 UnboundLocalError。

解决方案：初始化变量

为了解决这个问题，在 try 块之前初始化变量 a，为其赋予一个默认值。这样，即使 try 块中的赋值失败，a 仍然有一个值，可以在 except 块中安全地引用。

def user_input():
    a = None  # 初始化 'a'
    try:
        a = int(input("Enter a number: "))
    except ValueError:
        print("\nInvalid input. Please enter a number.\n") # 修改打印信息，因为a的值可能为None
    else:
        return a

在这个修改后的版本中，a 被初始化为 None。如果用户输入无效，except 块将执行，并且 a 可以被安全地引用（尽管它的值为 None）。 同时，也需要修改except 块中的打印信息，因为此时 a 的值可能为 None。

替代方案：循环提示输入

如果希望用户必须输入一个有效的数字，可以使用循环来重复提示用户输入，直到输入有效为止。

def user_input():
    while True:
        try:
            a = int(input("Enter a number: "))
            return a
        except ValueError:
            print("\nInvalid input. Please enter a number.\n")

这个版本使用一个无限循环，直到用户输入一个有效的整数。如果输入无效，ValueError 异常将被捕获，并向用户显示错误消息。循环会继续，直到用户输入有效的整数，然后该整数将被返回。

总结

处理用户输入时，需要格外小心，因为用户可能会输入任何内容。通过初始化变量或使用循环来验证输入，可以编写更健壮的程序，避免意外的错误和崩溃。在处理 int() 函数的 ValueError 异常时，确保在 try 块之前初始化变量，或者使用循环来强制用户输入有效的数字。