Question

python的新手...... 我有以下类Key，它扩展了dict：

class Key( dict ):

    def __init__( self ):
        self = { some dictionary stuff... }

    def __getstate__(self):
        state = self.__dict__.copy()
        return state

    def __setstate__(self, state):
        self.__dict__.update( state )

我想使用pickle.dump使用其数据保存类的实例，然后使用pickle.load检索数据。我知道我应该以某种方式改变getstate和setstate，但是，我并不完全清楚我应该怎么做...任何帮助都将不胜感激！

Answer 1

我写了一个dict的子类，它就是这样做的。

class AttrDict(dict):
    """A dictionary with attribute-style access. It maps attribute access to
    the real dictionary.  """
    def __init__(self, *args, **kwargs):
        dict.__init__(self, *args, **kwargs)

    def __getstate__(self):
        return self.__dict__.items()

    def __setstate__(self, items):
        for key, val in items:
            self.__dict__[key] = val

    def __repr__(self):
        return "%s(%s)" % (self.__class__.__name__, dict.__repr__(self))

    def __setitem__(self, key, value):
        return super(AttrDict, self).__setitem__(key, value)

    def __getitem__(self, name):
        return super(AttrDict, self).__getitem__(name)

    def __delitem__(self, name):
        return super(AttrDict, self).__delitem__(name)

    __getattr__ = __getitem__
    __setattr__ = __setitem__

    def copy(self):
        return AttrDict(self)

它基本上将状态转换为基本元组，然后再将其转换为unpickle。

但请注意，您必须拥有可用于unpickle的原始源文件。酸洗实际上并不保存类本身，只保存实例状态。 Python将需要原始的类定义来重新创建。

如何在Python中使用pickle创建持久化类

1 个答案: