Как создать постоянный класс, используя pickle в Python - PullRequest
Новая
       31

Как создать постоянный класс, используя pickle в Python

1 голос
/ 23 февраля 2012

Новое в 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, однако, не совсем ясно, как я должен это делать ... любая помощь будет принята с благодарностью!

1 Ответ

3 голосов
/ 23 февраля 2012

Я написал подкласс 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)

Он в основном преобразует состояние в базовый кортеж и возвращает его обратно, чтобы распечатать.

Но имейте в виду, что вам нужно иметь исходный файл, доступный для распаковки. Травление фактически не сохраняет сам класс, а только состояние экземпляра. Python потребуется исходное определение класса для повторного создания из.

...