41

我有一个嵌套类:

类小部件类型(对象):
    
    类浮动类型(对象):
        经过
    
    类文本类型(对象):
        经过

..和一个像这样引用嵌套类类型(不是它的实例)的对象

类 ObjectToPickle(对象):
     def __init__(self):
         self.type = WidgetType.TextType

尝试序列化 ObjectToPickle 类的实例会导致:

PicklingError:无法腌制 <class 'setmanager.app.site.widget_data_types.TextType'>

有没有办法在 python 中腌制嵌套类?

4

7 回答 7

32

我知道这是一个非常古老的问题,但除了重新构建代码的明显且很可能是正确的答案之外,我从未明确看到此问题的令人满意的解决方案。

不幸的是,做这样的事情并不总是可行的,在这种情况下,作为最后的手段,可以腌制在另一个类中定义的类的实例。

__reduce__函数的 python 文档指出您可以返回

将被调用以创建对象的初始版本的可调用对象。元组的下一个元素将为这个可调用对象提供参数。

因此,您所需要的只是一个可以返回相应类的实例的对象。此类本身必须是可腌制的(因此,必须存在于该__main__级别上),并且可以简单如下:

class _NestedClassGetter(object):
    """
    When called with the containing class as the first argument, 
    and the name of the nested class as the second argument,
    returns an instance of the nested class.
    """
    def __call__(self, containing_class, class_name):
        nested_class = getattr(containing_class, class_name)
        # return an instance of a nested_class. Some more intelligence could be
        # applied for class construction if necessary.
        return nested_class()

因此,剩下的就是__reduce__在 FloatType 的方法中返回适当的参数:

class WidgetType(object):

    class FloatType(object):
        def __reduce__(self):
            # return a class which can return this class when called with the 
            # appropriate tuple of arguments
            return (_NestedClassGetter(), (WidgetType, self.__class__.__name__, ))

结果是一个嵌套的类,但实例可以腌制(需要进一步的工作来转储/加载信息,但根据文档__state__,这相对简单)。__reduce__

同样的技术(稍加修改代码)可以应用于深度嵌套的类。

一个完整的例子:

import pickle


class ParentClass(object):

    class NestedClass(object):
        def __init__(self, var1):
            self.var1 = var1

        def __reduce__(self):
            state = self.__dict__.copy()
            return (_NestedClassGetter(), 
                    (ParentClass, self.__class__.__name__, ), 
                    state,
                    )


class _NestedClassGetter(object):
    """
    When called with the containing class as the first argument, 
    and the name of the nested class as the second argument,
    returns an instance of the nested class.
    """
    def __call__(self, containing_class, class_name):
        nested_class = getattr(containing_class, class_name)

        # make an instance of a simple object (this one will do), for which we can change the
        # __class__ later on.
        nested_instance = _NestedClassGetter()

        # set the class of the instance, the __init__ will never be called on the class
        # but the original state will be set later on by pickle.
        nested_instance.__class__ = nested_class
        return nested_instance



if __name__ == '__main__':

    orig = ParentClass.NestedClass(var1=['hello', 'world'])

    pickle.dump(orig, open('simple.pickle', 'w'))

    pickled = pickle.load(open('simple.pickle', 'r'))

    print type(pickled)
    print pickled.var1

我对此的最后一点是要记住其他答案所说的话:

如果您有能力这样做,请考虑重新分解您的代码以避免首先出现嵌套类。

于 2012-07-15T16:50:45.903 回答
30

pickle 模块正在尝试从模块中获取 TextType 类。但是由于该类是嵌套的,因此它不起作用。jasonjs 的建议会奏效。以下是 pickle.py 中导致错误消息的行:

    try:
        __import__(module)
        mod = sys.modules[module]
        klass = getattr(mod, name)
    except (ImportError, KeyError, AttributeError):
        raise PicklingError(
            "Can't pickle %r: it's not found as %s.%s" %
            (obj, module, name))

klass = getattr(mod, name)当然,在嵌套类的情况下不起作用。为了演示发生了什么,请尝试在腌制实例之前添加这些行:

import sys
setattr(sys.modules[__name__], 'TextType', WidgetType.TextType)

此代码将 TextType 作为属性添加到模块。酸洗应该工作得很好。不过,我不建议您使用此 hack。

于 2009-12-22T17:51:23.100 回答
6

如果您使用dill而不是pickle,它可以工作。

>>> import dill
>>> 
>>> class WidgetType(object):
...   class FloatType(object):
...     pass
...   class TextType(object):
...     pass
... 
>>> class ObjectToPickle(object):
...   def __init__(self):
...     self.type = WidgetType.TextType
... 
>>> x = ObjectToPickle()
>>> 
>>> _x = dill.dumps(x)
>>> x_ = dill.loads(_x)
>>> x_
<__main__.ObjectToPickle object at 0x10b20a250>
>>> x_.type
<class '__main__.TextType'>

在这里获取莳萝:https ://github.com/uqfoundation/dill

于 2014-01-25T02:09:42.337 回答
4

在 Sage ( www.sagemath.org ) 中,我们有很多这种酸洗问题的实例。我们决定系统地解决它的方法是将外部类放在一个特定的元类中,其目标是实现和隐藏黑客。请注意,如果有多个嵌套级别,这会自动通过嵌套类传播。

于 2010-03-06T10:41:46.163 回答
2

Pickle 仅适用于模块范围(顶级)中定义的类。在这种情况下,看起来您可以在模块范围内定义嵌套类,然后将它们设置为 WidgetType 上的属性,假设有理由不只是在代码中引用TextTypeFloatType。或者,导入他们所在的模块并使用widget_type.TextTypeand widget_type.FloatType

于 2009-12-22T17:36:12.227 回答
1

Nadia 的回答非常完整——这实际上不是你想做的事情;你确定你不能使用继承WidgetTypes而不是嵌套类吗?

使用嵌套类的唯一原因是将类紧密地封装在一起,您的具体示例对我来说看起来像是一个直接的继承候选者 - 将WidgetType类嵌套在一起没有任何好处;将它们放在一个模块中并从基础继承WidgetType

于 2009-12-22T21:19:04.663 回答
0

这似乎在较新版本的 Python 中运行良好。我在 v3.8 中尝试过,它能够腌制和解开嵌套类。

于 2021-10-09T14:43:43.830 回答