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

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

附Python Enum使用总结

1.枚举类不能用来实例化对象

2.访问枚举类中的某一项，直接使用类名访问加上要访问的项即可，比如 color.RED

3.枚举类里面定义的 Key = Value，在类外部不能修改 Value 值

4.枚举项可以用来比较，使用 ==，或者 is

5.枚举类中的 Key 不能相同，Value 可以相同，但是 Value 相同的各项 Key 都会当做别名

6.枚举类可以用 for 进行遍历，members.items() 可以遍历出含有别名的类

7.如果要枚举类中的 key 也不能相同，需要导入 unique 对枚举类进行装饰

来源：https://segmentfault.com/a/1190000041438811