关于python: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: 3>
>>> 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)
<enum 'Color'>
>>> 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

这显然大大限度了枚举类的扩大，一个不能拜访成员的函数，用途也就不大了。