Python Enum 使用的几点注意事项

Enum 是个类

所以

基本的类操作都可以用

也就是我们可以添加自己的方法

class Mood(Enum):
    FUNKY = 1
    HAPPY = 3

    def describe(self):
        # self is the member here
        return self.name, self.value

    def __str__(self):
        return 'my custom str! {0}'.format(self.value)

    @classmethod
    def favorite_mood(cls):
        # cls here is the enumeration
        return cls.HAPPY

>>> Mood.favorite_mood()

>>> Mood.HAPPY.describe()
('HAPPY', 3)
>>> str(Mood.FUNKY)
'my custom str! 1'

Enum的每个类成员,都会自动被转换成当前类的一个实例

from enum import Enum
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

>>> type(Color.RED)

>>> isinstance(Color.GREEN, Color)
True

这就意味着，我们不能用枚举成员直接当成它的value来用：

>>> Color.RED == 1
False
>>> Color.RED.value == 1
False

枚举成员还有个name属性，跟它的变量名相同

>>> Color.RED.name == "RED"
True

枚举类有个大坑：父类有成员的时候，就不能定义子类

所以，对于上面的Color类，如果还想定义一个子类，就会出错：

>>> class MoreColor(Color):
...     PINK = 17
...
Traceback (most recent call last):
...
TypeError: MoreColor: cannot extend enumeration 'Color'

但父类没有枚举成员，仅仅定义了函数是可以的：

class Foo(Enum):
    def some_behavior(self):
        pass

class Bar(Foo):
    HAPPY = 1
    SAD = 2

这显然大大限制了枚举类的扩展，一个不能访问成员的函数，用处也就不大了。