Pythonで複雑な列挙型をpickleする方法

Question

pickled複雑なenumインスタンスをunpickleしようとすると、常にValueError：BLUEは有効なColorsではありません。

どのようにピックルしてアンピッキングする方法はありますか？

from pickle import loads, dumps 
from enum import Enum 


class ColorType(object): 

    def __init__(self, counter, name): 
     self.counter = counter 
     self.name = name 

    def __str__(self): 
     return self.name 


class Colors(Enum): 
    GREEN = ColorType(1, 'GREEN') 
    BLUE = ColorType(2, 'BLUE') 


color = Colors.BLUE 
print(color is loads(dumps(color))) 

私はPython 2.7を使用しています。

Answer 1

列挙型の値としてカスタムクラスを使用しないでください。ここには必要ありません。まったく別のクラスを必要としないあなたの具体的な例は、あなただけ使用できます。

class Colors(Enum): 
    GREEN = 1 
    BLUE = 2 

    def __str__(self): 
     return self.name 

    @property 
    def counter(self): 
     return self.value 

これは良くstr()と.counter行動を持っています。コードでstr()をColor.<name>に直接送信するのではなく、Color.<name>.valueに適用する必要があります。

その他のカスタムメソッドと属性については、それらをサブクラスEnumに直接置くと、それらもenumメンバーの一部になります。エントリごとにさらに値が必要な場合は、タプルを設定し、__init__メソッドでそのタプルを引き離します。ドキュメントにはこれをさらに詳しく示すPlanet exampleがあります。

デモ：

>>> Colors.BLUE 
<Colors.BLUE: 2> 
>>> Colors.BLUE.value 
2 
>>> Colors.BLUE.counter 
2 
>>> str(Colors.BLUE) 
'BLUE' 
>>> Colors.BLUE is loads(dumps(Colors.BLUE)) 
True 

Answer 2

ここでの問題は、基本的な平等である：Colorsは、それが失敗しているColorType(2, 'BLUE')のアンピックル値の一致を見つけようとしているので

>>> ColorType(2, 'BLUE') == ColorType(2, 'BLUE') 
False 

。

解決策は単純です： `ColorTypeです 'に__eq__と__ne__メソッドを追加：

class ColorType(object): 

    def __init__(self, counter, name): 
     self.counter = counter 
     self.name = name 

    def __str__(self): 
     return self.name 

    def __eq__(self, other): 
     return self.name == other.name and self.counter == other.counter 

    def __ne__(self, other): 
     # not needed in Python 3 
     return self.name != other .name or self.counter != other.counter 

NB私はほとんどの場合、あなただけのEnum自体に必要な機能を追加する必要があることを@MartijnPietersに同意します。