假设我有一个通过其闭包递归的函数:
def outer():
def fact(n):
return 1 if n == 0 else n * fact(n - 1)
return fact
我现在想序列化函数并使用
types.FunctionType
重建它:import pickle, marshal, copyreg, types
def make_cell(value):
return (lambda: value).__closure__[0]
def make_function(*args):
return types.FunctionType(*args)
copyreg.pickle(types.CodeType,
lambda code: (marshal.loads, (marshal.dumps(code),)))
copyreg.pickle(type((lambda i=0: lambda: i)().__closure__[0]),
lambda cell: (make_cell, (cell.cell_contents,)))
copyreg.pickle(types.FunctionType,
lambda fn: (make_function, (fn.__code__, {}, fn.__name__, fn.__defaults__, fn.__closure__)))
buf = pickle.dumps(outer())
fn = pickle.loads(buf)
这适用于普通闭包,但使用
fact
会导致无限递归,因为 pickle
尝试在其闭包中序列化 fact
。在 pickle
中处理递归数据结构的常用方法是在构造和初始化之间记住对象,但 function
对象是不可变的,fn.__closure__
(元组)和单元对象也是如此:>>> cell = (lambda i=0: lambda: i)().__closure__[0]
>>> cell.cell_contents = 5
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: attribute 'cell_contents' of 'cell' objects is not writable
据推测,在普通代码中构造递归函数时,语言必须做类似的事情,因为在构造函数对象之前,它无法放入其闭包中。构建我缺少的递归函数有什么魔力吗?
最佳答案
闭包绑定(bind)到一个自由变量,而不是它的值。对于自引用闭包,Python 需要做的就是首先为自由 fact
名称创建一个闭包(尚未绑定(bind)到任何东西),使用闭包创建函数对象,然后将 fact
绑定(bind)到该对象。
因此,您需要将创建一个闭包和一个函数合并到同一个外部函数中,这样您就可以为函数将绑定(bind)到的名称创建一个闭包:
def create_closure_and_function(*args):
func = None
def create_function_closure():
return func
closure = create_function_closure.__closure__
func = types.FunctionType(*args[:-1] + [closure])
return func
为了使这项工作与 unpickling 一起工作,您必须遍历闭包参数(
args[-1]
)并检测递归的位置,然后用 create_function_closure.__closure__[0]
替换该项目,我想。关于python - 如何序列化递归函数?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/18214743/