Python 的 enum 模块提供安全、可读、类型友好的具名常量定义方式;相比普通变量,Enum 防止重复赋值、拼写错误和越界取值,支持迭代、比较、序列化及 IDE 补全。
Python 的 enum 模块提供了一种定义具名常量集合的规范方式,比用普通变量或字典更安全、可读性更强,也支持迭代、比较和序列化等操作。
直接用字符串或数字定义状态(如 STATUS_ACTIVE = 1)容易出错:可能重复赋值、拼写错误、无法限制取值范围。Enum 强制所有合法值必须显式声明,且每个成员是唯一的对象实例,天然支持类型检查和 IDE 自动补全。
Color.RED 和 State.RED 是独立的)is 判断、== 比较、in 成员检测继承 Enum 类即可创建枚举类。每个属性是一个枚举成员,自动获得唯一序号(从 1 开始),也可显式赋值。
from enum import Enumclass Color(Enum): RED = 1 GREEN = 2 BLUE = 3
访问方式
print(Color.RED) # Color.RED print(Color.RED.name) # 'RED' print(Color.RED.value) # 1 print(Color.RED.value == 1) # True print(Color(2)) # Color.GREEN(按 value 查找) print(Color['BLUE']) # Color.BLUE(按
name 查找)
IntEnum 允许与整数直接比较;StrEnum(Python 3.11+)让成员值默认为字符串且支持字符串比较;auto() 自动递增赋值,避免手动编号出错。
from enum import IntEnum, StrEnum, autoclass Priority(IntEnum): LOW = 10 MEDIUM = 20 HIGH = 30
print(Priority.HIGH > Priority.LOW) # True(IntEnum 支持数值比较)
class Status(StrEnum): PENDING = auto() # 自动设为 'PENDING' DONE = auto() # 自动设为 'DONE'
print(Status.PENDING == "PENDING") # True
枚举类本身可迭代,适合用于下拉选项或状态校验;转 JSON 需自定义编码器,通常推荐存 name 或 value 字段。
list(Color) 或 for c in Color:
value in Color.__members__.values() 或用 try/except 捕获 ValueError
import json
class Encoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, Enum):
return obj.name # 或 obj.value
return super().default(obj)
json.dumps({'color': Color.RED}, cls=Encoder) # '{"color": "RED"}'