Python链式异常：如何获取根异常中的局部变量？

问题背景 #

你编写的异常局部变量转储代码在无链式异常时正常工作，但处理链式异常（通过__cause__/__context__遍历）时，后续异常的栈帧frame.f_locals始终为空。

核心原因 #

Python会在栈帧退出（比如异常抛出后，对应函数执行完成）时自动清理栈帧内的局部变量以释放内存。链式异常中，后续异常对应的栈帧上下文已经结束，直接访问frame.f_locals时变量已被回收。

有效解决方法 #

1. 用traceback.extract_tb捕获局部变量副本 #

使用traceback.extract_tb并设置capture_locals=True，该参数会在栈帧被销毁前主动保存局部变量的副本，避免变量被回收：

import sys
import traceback

def dump_all_exception_locals():
    type_, value, tb = sys.exc_info()
    while value:
        # 提取栈帧并捕获局部变量
        frame_summaries = traceback.extract_tb(tb, capture_locals=True)
        for fs in frame_summaries:
            print(f"文件: {fs.filename} | 行号: {fs.lineno} | 函数: {fs.name}")
            print(f"局部变量: {fs.locals}\n")
        # 遍历链式异常
        value = value.__cause__ or value.__context__
        tb = value.__traceback__ if value else None

# 测试示例
try:
    try:
        num = 10
        text = "5"
        result = num + text  # 触发TypeError
    except TypeError as e:
        ctx_var = "chain_test"
        raise RuntimeError("触发链式异常") from e
except:
    dump_all_exception_locals()

2. 全局异常钩子捕获 #

如果需要全局范围内捕获所有异常并保留局部变量，可以自定义sys.excepthook：

import sys
import traceback

def custom_exception_handler(type_, value, tb):
    while value:
        frame_summaries = traceback.extract_tb(tb, capture_locals=True)
        for fs in frame_summaries:
            print(f"[{fs.filename}:{fs.lineno}] {fs.name}")
            print(f"局部变量: {fs.locals}\n")
        value = value.__cause__ or value.__context__
        tb = value.__traceback__ if value else None

sys.excepthook = custom_exception_handler

# 测试链式异常
try:
    try:
        a = 1
        b = "2"
        c = a + b
    except TypeError as e:
        x = "context_var"
        raise RuntimeError("链式异常示例") from e
except:
    sys.excepthook(*sys.exc_info())

注意事项 #

• 避免直接遍历traceback.walk_tb后访问frame.f_locals，这种方式无法应对栈帧已被清理的场景。
• capture_locals=True会保存局部变量的副本，可能占用额外内存，调试场景下使用更合适。

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