9

このオブジェクトがその属性の 1 つを介して自分自身を参照する場合、スロットを持つクラスからオブジェクトをピクルする正しい方法は何ですか? これは、私の現在の実装での簡単な例ですが、100% 正しいかどうかはわかりません。

import weakref
import pickle

class my_class(object):

    __slots__ = ('an_int', 'ref_to_self', '__weakref__')

    def __init__(self):
        self.an_int = 42
        self.ref_to_self = weakref.WeakKeyDictionary({self: 1})

    # How to best write __getstate__ and __setstate__?
    def __getstate__(self):

        obj_slot_values = dict((k, getattr(self, k)) for k in self.__slots__)
        # Conversion to a usual dictionary:
        obj_slot_values['ref_to_self'] = dict(obj_slot_values['ref_to_self'])
        # Unpicklable weakref object:
        del obj_slot_values['__weakref__']
        return obj_slot_values

    def __setstate__(self, data_dict):
        # print data_dict
        for (name, value) in data_dict.iteritems():
            setattr(self, name, value)
        # Conversion of the dict back to a WeakKeyDictionary:
        self.ref_to_self = weakref.WeakKeyDictionary(
            self.ref_to_self.iteritems())

これは、たとえば次のようにテストできます。

def test_pickling(obj):
    "Pickles obj and unpickles it.  Returns the unpickled object"

    obj_pickled = pickle.dumps(obj)
    obj_unpickled = pickle.loads(obj_pickled)

    # Self-references should be kept:
    print "OK?", obj_unpickled == obj_unpickled.ref_to_self.keys()[0]
    print "OK?", isinstance(obj_unpickled.ref_to_self,
                            weakref.WeakKeyDictionary)

    return obj_unpickled

if __name__ == '__main__':
    obj = my_class()
    obj_unpickled = test_pickling(obj)
    obj_unpickled2 = test_pickling(obj_unpickled)

これは正しい/堅牢な実装ですか? を持つクラスから継承された場合、どのように記述する必要が__getstate__ありますか? 「循環」口述のために内部にメモリリークがありますか？__setstate__my_class__slots____setstate__

PEP 307my_classには、オブジェクトのピクル化が堅牢な方法でまったく可能かどうか疑問に思う発言があります。

この__getstate__メソッドは、オブジェクト自体を参照せずに、オブジェクトの状態を表す picklable 値を返す必要があります。

これは、オブジェクト自体への参照がピクルされているという事実と衝突しますか?

たくさんの質問があります。コメント、コメント、アドバイスをいただければ幸いです。

4

1 に答える 1

6

元の投稿が示唆しているものは十分に機能しているようです。

PEP 307 の内容については、次のとおりです。

__getstate__ メソッドは、オブジェクト自体を参照することなく、オブジェクトの状態を表す picklable 値を返す必要があります。

__getstate__メソッドが (ピクルできない) 元のオブジェクトを指していない表現を単に返さなければならないことを意味するだけであることを理解しています。したがって、元の (ピクル化できない) オブジェクトへの参照が行われない限り、自身を参照するオブジェクトを返すことは問題ありません。

于 2010-06-13T08:35:17.117 に答える