如何让Python异常记录延迟函数创建与执行的双调用栈？

常规实现的Delay惰性计算封装存在一个非常典型的可观测性缺陷：延迟任务实际执行时抛出的异常，只会携带执行触发点的调用栈信息，完全丢失了Delay实例创建时的上下文——也就是错误参数实际传入的位置。比如示例中init_runtime传入了不存在的文件名，最终报错只会指向读文件的load_historical_data调用，不会关联到init_runtime里的传参逻辑，排查问题时需要全链路搜索Delay实例的创建位置，调试成本极高。

核心思路非常直接：在Delay实例初始化时就捕获当前的创建调用栈存为实例属性，执行任务时如果抛出异常，就把原始异常、创建栈、执行栈三类信息整合到自定义异常中统一抛出。
首先引入依赖、定义自定义异常类，再改造Delay实现：

import traceback

class LazyExecutionError(Exception):
    """延迟计算执行失败时抛出的自定义异常，携带完整双栈上下文"""
    def __init__(self, original_exc, creation_stack, execution_stack):
        self.original_exc = original_exc
        self.creation_stack = creation_stack
        self.execution_stack = execution_stack
        err_msg = (
            "惰性任务执行失败，完整上下文如下：\n"
            "--- 原始异常信息 ---\n"
            f"{type(original_exc).__name__}: {original_exc}\n"
            "--- Delay实例创建栈（错误传参位置参考） ---\n"
            f"{creation_stack}"
            "--- 任务实际执行栈 ---\n"
            f"{execution_stack}"
        )
        super().__init__(err_msg)

class Delay:
    def __init__(self, fn, *args, **kwargs):
        self.partial = (fn, args, kwargs)
        self.result = None
        # 捕获创建时的调用栈，切片去掉当前__init__方法本身的栈帧，避免冗余信息
        self.creation_stack = "".join(traceback.format_stack()[:-1])

    def __call__(self):
        if self.partial is not None:
            fn, args, kwargs = self.partial
            try:
                self.result = fn(*args, **kwargs)
            except Exception as e:
                execution_stack = traceback.format_exc()
                # 用raise from关联原始异常，保留Python原生异常链
                raise LazyExecutionError(e, self.creation_stack, execution_stack) from e
            self.partial = None
        return self.result

改造后再触发示例中的文件名错误时，异常信息会直接打印出init_runtime中创建Delay实例的栈帧，一眼就能定位到传参错误的位置。

除了上述预存格式化栈字符串的方案，还有两种常见实现路径，各自适用场景不同：

• 预存格式化栈字符串（上述基础方案）
  • 优点：实现无额外依赖，兼容所有Python解释器版本；异常信息直接可读，不需要额外调试操作；存储的是静态字符串，不会持有栈帧对象，不存在内存泄漏风险，适合生产环境使用
  • 缺点：栈信息是初始化时格式化好的静态文本，无法在交互式调试时直接探查创建点的局部变量、运行时状态，只能通过打印的文本排查问题；如果Delay实例创建后永远不被执行，预存的栈字符串会带来极小的固定内存开销，通常可以忽略
• 预存原始栈帧对象
  • 实现方式：初始化时通过inspect.stack()或sys._getframe()拿到原始栈帧对象存为实例属性，抛出异常时再动态格式化栈信息
  • 优点：保留了创建点完整的栈帧上下文，交互式调试时可以直接读取创建位置的局部变量、参数值、调用链上的对象状态，排查复杂逻辑问题时效率更高
  • 缺点：栈帧对象会持有对应作用域内所有变量的引用，导致这些对象无法被垃圾回收，如果Delay实例生命周期长、创建点作用域持有大内存对象，会造成明显的内存泄漏；sys._getframe属于CPython解释器的实现细节，不是Python语言标准规定的公开API，在PyPy、GraalPy等其他解释器上可能存在兼容性问题
• 全局追踪动态匹配
  • 实现方式：通过函数钩子、调试追踪工具全局监听Delay类的实例化行为，统一记录所有实例的创建位置，任务执行报错时再动态匹配对应实例的创建栈
  • 优点：不需要在Delay实例上存储任何栈相关属性，没有单实例的额外内存开销
  • 缺点：实现复杂度极高，需要依赖第三方追踪库，全局钩子会给整个程序带来固定的性能损耗，仅适合本地调试场景，完全不适合生产环境使用

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