__slots__ 的pickle问题

报错  TypeError: a class that defines __slots__ without defining __getstate__ cannot be pickled

>>> class B(object):
    __slots__ = ('aa', 'bb', 'cc')
    def __init__(self, *args):
        for i, item in enumerate(self.__slots__):
            setattr(self, item, args[i])

    def update(self, **kwargs):
        for k, v in kwargs.items():
            if k in self.__slots__:
                setattr(self, k, v)

    def __getstate__(self):
	    return dict([(it,getattr(self,it)) for it in self.__slots__])
	
    def __setstate__(self, d):
	    self.update(**d)

	    
>>> b = B(1,2,3)
>>> bp = pickle.dumps(b)
>>> bn = pickle.loads(bp)
>>> bn.aa, bn.bb, bn.cc
(1, 2, 3)
>>> dir(bn)
['__class__', '__delattr__', '__doc__', '__format__', '__getattribute__', '__getstate__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__setstate__', '__sizeof__', '__slots__', '__str__', '__subclasshook__', 'aa', 'bb', 'cc', 'update']


你可能感兴趣的:(__slots__,__getstate__,__setstate__)