使用示例
def foo(a):
def bar(b):
return a+b
return bar
d = {1:foo(1), 2:foo(2)}
似乎 pickle 模块不适用于未在模块范围内定义的函数,因此 pickle 'd' 将不起作用。我应该考虑另一种酸洗机制吗?
恐怕你不能腌制嵌套函数。
该pickle
模块按名称序列化函数。也就是说,如果您myfunc
在模块中有一个函数,mymodule
它只是保存名称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
但请注意,当您反序列化实例时,您需要有相同的类定义可用。
如果您使用dill
而不是pickle
.
>>> import dill
>>>
>>> def foo(a):
... def bar(b):
... return a+b
... return bar
...
>>> d = {1:foo(1), 2:foo(2)}
>>>
>>> _d = dill.dumps(d)
>>> d_ = dill.loads(_d)
>>> d_
{1: <function bar at 0x108cfe848>, 2: <function bar at 0x108cfe8c0>}
>>> d[1](0) + d[2](10)
13
>>>
根据 Blckknght 的回答。如果嵌套函数是唯一的外部序列化类型并将其用作decorator
,您可以functools.warps
在内部函数定义的顶部添加以引导其他解释找到正确的名称:
from functools import warps
def foo(func):
@wraps(func)
def bar(b):
return func(b)
return bar
@foo
def zzz(b):
return b