Python: pickle 嵌套函数

Question

使用示例

def foo(a):
    def bar(b):
        return a+b
    return bar

d = {1:foo(1), 2:foo(2)}

pickle 模块似乎无法使用未在模块范围内定义的函数，因此 pickling 'd' 将不起作用。我应该考虑另一种 pickle 机制吗？

Answer 1

恐怕你不能 pickle 嵌套函数。

pickle 模块按名称序列化函数。也就是说，如果您在模块 mymodule 中有一个函数 myfunc，它只会保存名称 mymodule.myfunc 并在反序列化时再次查找它。 (这是一个重要的安全性和兼容性问题，因为它保证反序列化代码使用它自己的函数定义，而不是可能被破坏或过时的原始定义。)

唉，pickle 不能用嵌套函数做到这一点，因为没有办法通过名称直接寻址它们。例如，您的 bar 函数无法从 foo 外部访问。

如果您需要一个像函数一样工作的可序列化对象，您可以使用 __call__ 方法创建一个类:

class foo(object):
    def __init__(self, a):
        self.a = a
    def __call__(self, b): # the function formerly known as "bar"
        return self.a + b

这就像问题中的嵌套函数一样工作，应该不会对 pickle 造成问题。但请注意，当您反序列化一个 foo 实例时，您需要有相同的类定义可用。