Python中如何调整异常抛出位置、移除自定义异常的多余栈信息？

核心原理：Python 异常对象的__traceback__属性存储了完整的回溯栈链，每个栈帧通过tb_next指向下一层更外层的调用栈帧，直接修改该属性即可跳过不需要的栈信息。

针对示例代码的改法 #

def func1(val):
    # 优化判断写法，无需和True做等值比较
    if isinstance(val, int):
        print('value:', val)
    else:
        err = TypeError('must be integer')
        # 跳过当前raise所在的最内层栈帧
        err.__traceback__ = err.__traceback__.tb_next
        raise err from None

def func2():
    func1(1)
    func1('1')

func2()

修改后运行输出：

value: 1
Traceback (most recent call last):
  File "c:/Users/sss/etc/exceptionTest/exceptionTest.py", line 15, in <module>
    func2()
  File "c:/Users/sss/etc/exceptionTest/exceptionTest.py", line 12, in func2
    func1('1')
TypeError: must be integer

已经完全移除了原示例中raise语句对应的冗余栈帧。

通用封装工具 #

如果需要在多处复用抛出异常的逻辑，可以封装公共方法，按需调整tb_next的调用次数适配场景：

def throw(exc):
    # 跳过当前工具函数、业务代码raise对应的两层栈帧
    exc.__traceback__ = exc.__traceback__.tb_next.tb_next
    raise exc from None

使用时替换原生raise语句即可：

# 原写法：raise TypeError('must be integer')
# 替换为：
throw(TypeError('must be integer'))

仅打印时过滤的场景 #

如果不需要修改原始异常对象，只是在捕获异常后打印时过滤指定栈帧，可以配合traceback模块自定义输出逻辑：

import traceback

try:
    func2()
except TypeError as e:
    # 跳过最内层的raise栈帧再打印
    filtered_tb = e.__traceback__.tb_next
    traceback.print_exception(type(e), e, filtered_tb)

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