如何有選擇地腌制類實例的變量？

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,)