用Python腌制静态方法


问题内容

我一直在尝试腌制一个包含对静态类方法的引用的对象。泡菜失败(例如module.MyClass.foo),表示无法进行泡菜(因为module.foo不存在)。
我提出了以下解决方案,使用包装对象在调用时定位函数,保存容器类和函数名称:

class PicklableStaticMethod(object):
    """Picklable version of a static method.
    Typical usage:
        class MyClass:
            @staticmethod
            def doit():
                print "done"
        # This cannot be pickled:
        non_picklable = MyClass.doit
        # This can be pickled:
        picklable = PicklableStaticMethod(MyClass.doit, MyClass)
    """
    def __init__(self, func, parent_class):
        self.func_name = func.func_name
        self.parent_class = parent_class
    def __call__(self, *args, **kwargs):
        func = getattr(self.parent_class, self.func_name)
        return func(*args, **kwargs)

我想知道,是否有更好的-更标准的方法-
腌制这样的物体?我不想对全局pickle过程进行更改(copy_reg例如使用),但是以下模式将非常有用:class
MyClass(object):@picklable_staticmethod def foo():打印“完成”。

我的尝试没有成功,特别是因为我无法从foo函数中提取所有者类。我什至愿意满足显式规范(例如@picklable_staticmethod(MyClass))的要求,但是我不知道MyClass在定义类时直接引用该类的任何方法。

任何想法都很棒!

约纳坦


问题答案:

这似乎有效。

class PickleableStaticMethod(object):
    def __init__(self, fn, cls=None):
        self.cls = cls
        self.fn = fn
    def __call__(self, *args, **kwargs):
        return self.fn(*args, **kwargs)
    def __get__(self, obj, cls):
        return PickleableStaticMethod(self.fn, cls)
    def __getstate__(self):
        return (self.cls, self.fn.__name__)
    def __setstate__(self, state):
        self.cls, name = state
        self.fn = getattr(self.cls, name).fn

诀窍是在从类中获得静态方法时,将其锁定。

备选方案:您可以使用元类化为所有静态方法.__parentclass__赋予属性。然后,您可以子类化,Pickler并为每个子类实例提供其自己的.dispatch表,然后可以对其进行修改而不会影响全局调度表(Pickler.dispatch)。进行酸洗,解酸和调用方法可能会更快一些。