Как перезаписать методы дампа/загрузки в классе рассола — настройка травления и распаковки — Python

Вы хотите настроить способ выделения и извлечения состояния объекта, а не настраивать функции загрузки и выгрузки.

Вам придется изучить маринование и рассол главы об экземплярах классов, в вашем случае должно быть достаточно определения методов __getstate__ и __setstate__.

Что происходит в вашем случае, так это то, что существует атрибут уровня класса с экземплярами EnumValue, которые должны быть константами. Но при распаковке создаются новые экземпляры EnumValue, которые больше не связаны с атрибутом уровня класса.

Экземпляры EnumValue имеют атрибут index, который вы можете использовать для захвата их состояния в виде целого числа вместо экземпляра EnumValue, который мы можем использовать, чтобы снова найти правильную константу при восстановлении ваших экземпляров:

 class CellSizeRelation(Option):
     # skipping your enum definition and __init__ here

     def __getstate__(self):
         # capture what is normally pickled
         state = self.__dict__.copy()
         # replace the `value` key (now an EnumValue instance), with it's index:
         state['value'] = state['value'].index
         # what we return here will be stored in the pickle
         return state

     def __setstate__(self, newstate):
         # re-create the EnumState instance based on the stored index
         newstate['value'] = self.Values[newstate['value']]
         # re-instate our __dict__ state from the pickled state
         self.__dict__.update(newstate)

Итак, обычно, если __getstate__ нет, экземпляр __dict__ травится. Теперь мы возвращаем копию этого __dict__, но мы заменили экземпляр EnumValue на его индекс (простое целое число). При распаковке обычно новый экземпляр __dict__ обновляется с помощью нераспакованного __dict__, который мы зафиксировали при травлении, но теперь, когда мы определили __setstate__, мы можем снова поменять индекс перечисления на правильный EnumValue.

EnumValue зависит от id идентичности между Enum объектами "типа перечисления". Это имеет некоторые преимущества и недостатки.

Основное преимущество состоит в том, что два вызова Enum('A', 'B') определяют разные типы перечисления. Так:

osx = Enum('Jaguar', 'Tiger', 'Leopard')
bigcats = Enum('Jaguar', 'Tiger', 'Leopard')

Если вы хотите отличить OS X 10.4 от полосатой машины для убийств, это может быть полезно.

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

Итак, ваше решение не может включать в себя какой-либо хакерский рассол; придется взломать модуль enum.

Вам нужно будет определить разумный __cmp__ метод для Enum, который делает то, что имеет для вас смысл. Если вы можете отказаться от различия osx-vs.-bigcats, это легко. Если вы не можете, вам нужен какой-то другой способ (может быть, добавление явного имени тега в определение enum или необязательный, но в противном случае неявно-инкрементный счетчик?), который обрабатывает это.