如何有选择地腌制类实例的变量？

Question

我有一个类实例：

class A:
    def __init__(self,shape):
        self.shape = shape
        self.matrix = numpy.zeros(shape)

我对self.matrix做了一些乘法。 当我腌制A类的实例时，由于某种原因，我不想腌制self.matrix 。

目前，我的解决方案是在腌制之前将self.matrix = None设置为self.matrix = None ，并在从腌制文件中加载self.matrix重置zeros(shape) 。

有没有更优雅的解决方案？ 就像Java中的transient关键字一样。

Answer 1

使用钩子方法可以限制腌制和未腌制的物质。

在这里，您可以使用__getstate__和相应的__setstate__方法：

class A:
    def __init__(self, shape):
        self.shape = shape
        self.matrix = numpy.zeros(shape)

    def __getstate__(self):
        return (self.shape,)

    def __setstate__(self, state):
        self.shape, = state
        self.matrix = numpy.zeros(self.shape)

可以将__setstate__简化为：

    def __setstate__(self, state):
        self.__init__(*state)

如果您的__init__以这种方式可重复使用。 在这种情况下，您也可以使用__getinitargs__代替__getstate__并将__setstate__ ：

class A:
    def __init__(self, shape):
        self.shape = shape
        self.matrix = numpy.zeros(shape)

    def __getinitargs__(self):
        return (self.shape,)