Python сериализует лексические замыкания?
есть ли способ сериализовать лексическое закрытие в Python с помощью стандартной библиотеки? Пикл и маршал, похоже, не работают с лексическими закрытиями. Я действительно не забочусь о деталях двоичной и Строковой сериализации и т. д. это просто должно сработать. Например:
def foo(bar, baz) :
def closure(waldo) :
return baz * waldo
return closure
Я хотел бы просто иметь возможность сбрасывать экземпляры закрытия в файл и читать их обратно.
изменить: Один относительно очевидный способ, которым это может быть решено, - это некоторые хаки отражения преобразование лексических замыканий в объекты класса и наоборот. Можно затем преобразовать в классы, сериализовать, восстановить, преобразовать обратно к затворам. Черт возьми, учитывая, что Python типизирован уткой, если вы перегружаете оператор вызова функции класса, чтобы он выглядел как функция, вам даже не нужно будет преобразовывать его обратно в закрытие, и код, использующий его, не будет знать разницы. Если есть какие-либо гуру API отражения Python, пожалуйста, говорите.
5 ответов
если вы просто используете класс __call__
метод для начала, все должно работать гладко с pickle
.
class foo(object):
def __init__(self, bar, baz):
self.baz = baz
def __call__(self,waldo):
return self.baz * waldo
С другой стороны, хак, который преобразовал закрытие в экземпляр нового класса, созданного во время выполнения, не будет работать из-за способа pickle
имеет дело с классами и экземплярами. pickle
не хранит классы; только имя модуля и имя класса. При чтении экземпляра или класса он пытается импортировать модуль и найти в нем необходимый класс. Если использовать класс, созданный на лету, вам не повезло.
PiCloud выпустила Пиклер с открытым исходным кодом (LGPL), который может обрабатывать закрытие функций и многое другое полезное. Его можно использовать независимо от их инфраструктуры облачных вычислений - это просто обычный Пиклер. Подробно описана здесь, и вы можете загрузить код через "pip install cloud". В любом случае, он делает то, что ты хочешь. Давайте продемонстрируем это, замариновав закрытие:
import pickle
from StringIO import StringIO
import cloud
# generate a closure
def foo(bar, baz):
def closure(waldo):
return baz * waldo
return closure
closey = foo(3, 5)
# use the picloud pickler to pickle to a string
f = StringIO()
pickler = cloud.serialization.cloudpickle.CloudPickler(f)
pickler.dump(closey)
#rewind the virtual file and reload
f.seek(0)
closey2 = pickle.load(f)
теперь у нас есть closey
, первоначальное закрытие, и closey2
, тот, который был восстановлен из сериализации строк. Давайте проверим их.
>>> closey(4)
20
>>> closey2(4)
20
красиво. Модуль-это чистый python-вы можете открыть его и легко увидеть, что заставляет магию работать. (Ответ-много кода.)
да! Я получил его (по крайней мере, я думаю) - то есть более общую проблему маринования функции. Python настолько замечательный :), я узнал большую часть этого, хотя функция dir () и несколько веб-поисков. Также замечательно, что он [надеюсь] решен, мне это тоже нужно.
Я не сделал много тестов на то, насколько надежна эта вещь co_code (вложенные fcns и т. д.), и было бы неплохо, если бы кто-то мог посмотреть, как подключить Python, чтобы функции могли автоматически мариноваться (например, они иногда может быть закрытие аргументы).
Cython module _pickle_fcn.pyx
# -*- coding: utf-8 -*-
cdef extern from "Python.h":
object PyCell_New(object value)
def recreate_cell(value):
return PyCell_New(value)
файл Python
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# author gatoatigrado [ntung.com]
import cPickle, marshal, types
import pyximport; pyximport.install()
import _pickle_fcn
def foo(bar, baz) :
def closure(waldo) :
return baz * waldo
return closure
# really this problem is more about pickling arbitrary functions
# thanks so much to the original question poster for mentioning marshal
# I probably wouldn't have found out how to serialize func_code without it.
fcn_instance = foo("unused?", -1)
code_str = marshal.dumps(fcn_instance.func_code)
name = fcn_instance.func_name
defaults = fcn_instance.func_defaults
closure_values = [v.cell_contents for v in fcn_instance.func_closure]
serialized = cPickle.dumps((code_str, name, defaults, closure_values),
protocol=cPickle.HIGHEST_PROTOCOL)
code_str_, name_, defaults_, closure_values_ = cPickle.loads(serialized)
code_ = marshal.loads(code_str_)
closure_ = tuple([_pickle_fcn.recreate_cell(v) for v in closure_values_])
# reconstructing the globals is like pickling everything :)
# for most functions, it's likely not necessary
# it probably wouldn't be too much work to detect if fcn_instance global element is of type
# module, and handle that in some custom way
# (have the reconstruction reinstantiate the module)
reconstructed = types.FunctionType(code_, globals(),
name_, defaults_, closure_)
print(reconstructed(3))
спасибо,
Николас!--3-->
редактировать - более надежная глобальная обработка необходима для реальных случаев. fcn.func_code.co_names перечисляет глобальные имена.
Рецепт 500261: Названные Кортежи содержит функцию, которая определяет класс на лету. И этот класс поддерживает маринование.
вот суть:
result.__module__ = _sys._getframe(1).f_globals.get('__name__', '__main__')
в сочетании с предложение@Greg Ball создать новый класс во время выполнения он может ответить на ваш вопрос.
#!python
import marshal, pickle, new
def dump_func(f):
if f.func_closure:
closure = tuple(c.cell_contents for c in f.func_closure)
else:
closure = None
return marshal.dumps(f.func_code), f.func_defaults, closure
def load_func(code, defaults, closure, globs):
if closure is not None:
closure = reconstruct_closure(closure)
code = marshal.loads(code)
return new.function(code, globs, code.co_name, defaults, closure)
def reconstruct_closure(values):
ns = range(len(values))
src = ["def f(arg):"]
src += [" _%d = arg[%d]" % (n, n) for n in ns]
src += [" return lambda:(%s)" % ','.join("_%d"%n for n in ns), '']
src = '\n'.join(src)
try:
exec src
except:
raise SyntaxError(src)
return f(values).func_closure
if __name__ == '__main__':
def get_closure(x):
def the_closure(a, b=1):
return a * x + b, some_global
return the_closure
f = get_closure(10)
code, defaults, closure = dump_func(f)
dump = pickle.dumps((code, defaults, closure))
code, defaults, closure = pickle.loads(dump)
f = load_func(code, defaults, closure, globals())
some_global = 'some global'
print f(2)