Python枚举类enum

枚举 enum

枚举可以用来表示有限的几个值，如果我们想要让某个变量或者属性限定在某个值里，可以使用枚举类。
通常枚举成员我们建议使用大写。
下面来看一个实例:

from enum import Enum, unique

@unique
class Status(Enum):
    """
    定义一个状态枚举类,使用装饰器，避免枚举值重复，如果不需要刻意去掉装饰器
    """
    HIDE = 0
    SHOW = 1

# 如果需要,可以新增属性
Status.HIDE.label = '隐藏'
Status.SHOW.label = '显示'

# 直接用
print(type(Status.HIDE))
print(Status.HIDE, Status['HIDE'])
print(Status.HIDE.name)   # name 和 value是内置属性，不允许修改
print(Status.HIDE.value)
print(Status.HIDE.label)

# 循环遍历
for status in Status:
    print(status.name, status.value, status.label)
# 比较
print(Status.HIDE == Status.SHOW)

枚举适用于数据库某些字段，可以将字段中值限定在某些范围。例如从上面的枚举类中

# 假设数据库中的状态字段 status 只允许 0 或者 1 ，我们可以再查数据出来后，对其转换成枚举，
# 对于不存在的值，会直接抛出异常
t = Status(0)
if t == Status.SHOW:
    print("ok")
else:
    print("no")
print(t.name)
print(t.value)
print(t.label)

总结:

1. 装饰器unique不是必须的
2. 枚举成员建议大写
3. 枚举之间可以用is或者==比较
4. 可以使用 for in 来遍历枚举成员