如何在直接用int包裹input函数时优雅处理Python异常？

先看这段有问题的代码：

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

当用户输入非数字内容（比如v）时，会触发两层错误：

ValueError: invalid literal for int() with base 10: 'v'
During handling of the above exception, another exception occurred:
UnboundLocalError: local variable 'a' referenced before assignment

原因很直白：int(input(...))执行失败时，a根本没被赋值，所以except块里引用a自然会报错。

而分开调用input和int的写法就能正常运行：

a = input("Enter an integer: ")
try:
    a = int(a)
except ValueError:
    print(f"\n{a} is invalid.\n")
else:
    print(a)

但我希望直接用int(input())的紧凑形式完成类型转换，同时优雅处理异常，请问有没有替代方案？还是必须分开写？

当然有替代方案，不用拆分input和int也能实现需求，这里提供几种实用思路：

方案1：从异常对象中提取无效输入 #

ValueError的异常信息里包含了无效的输入内容，我们可以直接从异常对象里解析出来：

def user_input():
    try:
        return int(input("Enter a number: "))
    except ValueError as e:
        # 从异常消息中提取输入的字符串
        invalid_input = e.args[0].split(": ")[-1].strip("'")
        print(f"\n{invalid_input} is invalid.\n")

这种方式不需要单独存储input的返回值，直接从异常对象里拿到无效输入内容，保持了代码的紧凑性。

方案2：使用海象运算符（Python 3.8+） #

如果你的Python版本是3.8及以上，海象运算符:=是最完美的解决方案——它能在表达式内部完成赋值，既保持了int(input())的紧凑写法，又能在except块里正常引用输入值：

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

这里的执行逻辑是：先执行input()获取用户输入并赋值给a，再把a传给int()做转换。如果转换失败，a已经被赋值，所以except块里可以直接用它打印错误信息。

总结 #

• 若使用Python 3.8+，优先选海象运算符的方案，完全符合你想要的紧凑写法；
• 若版本较低，可选择从异常对象提取输入的方案；
• 当然，分开写input和int的方案虽然不够紧凑，但胜在兼容性好、逻辑直观，也是完全可行的选择。

内容的提问来源于stack exchange，提问作者Uchenna Adubasim