Python：如何通过函数使用Shelve/Dill存储工作区变量？

要实现把调用者函数里的所有变量存入shelve，我们可以借助Python的inspect模块获取调用栈的上下文，直接提取调用者的局部变量。下面是具体的实现思路和代码，完美适配你的调试需求：

核心函数实现 #

import shelve
import inspect

def save_caller_vars(shelve_path="shelve.out"):
    # 获取调用者的栈帧（当前函数是第0层，调用它的函数是第1层）
    caller_frame = inspect.currentframe().f_back
    try:
        # 提取调用者的全部局部变量
        caller_vars = caller_frame.f_locals
        # 写入shelve文件
        with shelve.open(shelve_path) as db:
            for var_name, var_value in caller_vars.items():
                db[var_name] = var_value
        print(f"调用者变量已成功保存到 {shelve_path}")
    finally:
        # 手动删除栈帧引用，避免内存泄漏
        del caller_frame

使用示例 #

把这个函数插入到你耗时代码的前序位置就行，比如：

def time_consuming_function():
    junk1 = "debug_value_1"
    junk2 = [10, 20, 30]
    junk3 = {"status": "pending", "data": {}}
    
    # 在耗时代码前保存变量
    save_caller_vars()
    
    # 下面是你不想重复跑的耗时代码
    # ... 你的业务逻辑 ...

运行后，shelve.out里就会包含junk1、junk2、junk3这些变量了。

加载变量调试 #

之后你可以单独写一段加载代码，直接复现当时的变量环境：

import shelve

with shelve.open("shelve.out") as db:
    # 把保存的变量加载到当前环境
    globals().update(db)
    # 现在可以直接调试这些变量了
    print(junk1)
    print(junk2)
    print(junk3)

一些注意点 #

• 不是所有对象都能被shelve序列化（比如某些自定义的非可序列化类），如果遇到报错，可以手动过滤掉这类变量，或者尝试用dill替代默认的序列化方式。
• 一定要在finally里删除栈帧引用，不然频繁调用这个函数可能会导致内存泄漏。
• 如果调用者是类的方法，f_locals里会包含self，你可以根据需要决定是否保留这个变量。

这样你就不用反复跑那段耗时极久的代码，直接加载保存的变量就能排查问题啦。

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