如何无需导入原模块即可解包用dill Pickle的无副作用函数foo？

这个问题我之前也碰到过——默认情况下，dill（和pickle）序列化函数时，其实只保存了函数的模块路径和名称，反序列化时会尝试导入原模块去找这个函数，所以没有my_project的环境自然会报错。好在你的函数无副作用、无外部依赖，完全可以直接序列化它的源代码或字节码，绕开模块依赖的问题，下面是几种靠谱的解决方案：

方法一：用dill直接保存/加载函数源代码 #

dill自带了处理函数源代码的工具，非常适合你的场景：

序列化步骤 #

import dill
from my_project.my_functions import foo

# 方式1：把源代码保存到单独的py文件
dill.source.save_source(foo, "foo_source.py")

# 方式2：把源代码字符串序列化到pkl文件
foo_source = dill.source.getsource(foo)
with open("foo.pkl", "wb") as f:
    dill.dump(foo_source, f)

反序列化步骤 #

import dill

# 对应方式1：从py文件加载函数
foo = dill.source.load_source("foo_source.py")

# 对应方式2：从pkl文件加载源代码并执行
with open("foo.pkl", "rb") as f:
    foo_source = dill.load(f)

# 执行源代码定义函数，拿到函数对象
local_namespace = {}
exec(foo_source, globals(), local_namespace)
foo = local_namespace["foo"]

方法二：手动序列化函数的核心组件 #

如果你想更底层地控制序列化过程，可以提取函数的代码对象、名称、默认参数等核心部分，反序列化时重新构建函数：

序列化步骤 #

import dill
import types
from my_project.my_functions import foo

# 提取函数的关键属性（无依赖的情况下这些就够了）
func_code = foo.__code__
func_name = foo.__name__
func_defaults = foo.__defaults__
func_kwdefaults = foo.__kwdefaults__

# 序列化这些属性
with open("foo.pkl", "wb") as f:
    dill.dump((func_code, func_name, func_defaults, func_kwdefaults), f)

反序列化步骤 #

import dill
import types

with open("foo.pkl", "rb") as f:
    func_code, func_name, func_defaults, func_kwdefaults = dill.load(f)

# 用types.FunctionType重新构建函数
foo = types.FunctionType(
    func_code,
    globals(),  # 因为函数无外部依赖，用当前全局命名空间即可
    func_name,
    func_defaults,
    func_kwdefaults
)

方法三：用cloudpickle一键解决（推荐） #

如果可以安装额外库的话，cloudpickle是专门为跨环境序列化设计的，它会自动处理无依赖函数的序列化，不需要手动操作：

序列化步骤 #

import cloudpickle
from my_project.my_functions import foo

with open("foo.pkl", "wb") as f:
    cloudpickle.dump(foo, f)

反序列化步骤 #

import cloudpickle

with open("foo.pkl", "rb") as f:
    foo = cloudpickle.load(f)

# 直接调用foo就行，完全不需要导入my_project

注意事项 #

• 一定要确保你的函数真的无外部依赖：比如没有引用my_project里的其他变量、类，也没有使用闭包捕获外部变量，否则反序列化时会因为找不到这些依赖报错。
• 源代码方式的兼容性更好，字节码方式（方法二）需要保证序列化和反序列化的Python版本一致，否则可能出现字节码不兼容的问题。

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