枚举 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)
总结:
- 装饰器unique不是必须的
- 枚举成员建议大写
- 枚举之间可以用is或者==比较
- 可以使用 for in 来遍历枚举成员