枚举类输出 django 的 choices 数据
在 django 项目的 models 中难免会使用 choices，例如用户的状态等。而在其他地方需要筛选用户时，使用枚举类型更加直观方便，为了统一管理，希望只定义一个枚举类型。
定义枚举
from enum import Enum
class EnumBase(Enum):
"""枚举基类"""
def __new__(cls, *args):
"""将定义的属性拆分，不影响 value 的正常使用"""
obj = object.__new__(cls)
if len(args) > 1:
obj._value_ = args[0] # 实际值还是给 value 使用
obj.des = args[1]
else:
obj._value_ = args[0]
return obj
@classmethod
def to_tuple(cls):
return tuple([(_.value, _.des) for _ in cls])
class UserStatus(EnumBase):
"""用户状态"""
# 为了在 django 的 choices 中使用，属性值为一个元组
REGISTER = (0, '待审核')
NORMAL = (1, '正常')
DISABLE = (2, '不可用')
DELETE = (3, '删除')
可以直接通过UserStatus.to_tuple()获取到 choices 值，并且也可以正常使用UserStatus.NORMAL.value获取属性值1。