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

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。

但请注意，您必须要有原始源文件才能解压。酸洗实际上并不保存类本身，只保存实例状态。 Python 需要重新创建原始类定义。