python IntEnum

# 聊聊Python里的IntEnum给常量一个体面的身份在Python里处理常量或者状态码的时候很多人习惯直接用数字或者字符串。比如写个status 1表示成功status 0表示失败。刚开始这么写挺方便的但项目稍微大一点问题就来了。你看着代码里的if user_type 2:得翻半天文档才能想起来这个2到底代表什么用户类型。更麻烦的是别人接手你的代码时看到满屏幕的魔法数字估计心里已经在默默吐槽了。这时候就该IntEnum出场了。它到底是什么IntEnum是Python标准库enum模块里的一个特殊枚举类。简单说它让你能用有意义的名称来代替那些赤裸裸的数字同时这些名称本质上还是整数。想象一下你有个抽屉专门放各种螺丝刀。以前你是直接记住“左边第三个抽屉第二格是十字螺丝刀”现在你给每个格子贴上标签“十字螺丝刀”、“一字螺丝刀”、“六角螺丝刀”。IntEnum就是那些标签让代码里的数字有了名字。它继承自Enum和int所以它既有枚举的特性成员不可变、可迭代、可比较又有整数的特性可以直接参与数学运算、可以当索引用。这种双重身份让它用起来特别顺手。它能解决什么问题最直接的用途就是消除魔法数字。代码里不再出现意义不明的数字取而代之的是Status.SUCCESS、UserType.ADMIN这样的表达一看就知道什么意思。它还能提供类型安全。虽然Python是动态类型语言但用了IntEnum之后你在代码里传递的就不再是普通的整数而是有明确含义的枚举值。这在一定程度上避免了传错值的可能——毕竟UserType.ADMIN和Permission.ADMIN虽然值可能相同但它们是不同的类型。另一个好处是集中管理。所有相关的常量都放在一个枚举类里修改起来特别方便。如果需要调整某个值只需要改一个地方不用在整个项目里搜索替换。怎么用起来用IntEnum其实特别简单。先导入然后像定义类一样定义你的枚举fromenumimportIntEnumclassColor(IntEnum):RED1GREEN2BLUE3现在你就有了一组颜色常量。可以用Color.RED来引用它它的值是1但比单纯的1有意义多了。枚举成员之间可以比较大小因为它们是整数。Color.RED Color.GREEN会返回True。你也可以直接拿它们做算术运算不过实际开发中这么用的场景不多。有时候需要根据值来获取枚举成员可以用Color(1)这会返回Color.RED。如果值不存在会抛出ValueError。遍历所有成员也很简单直接对枚举类做for循环就行。这在生成下拉菜单选项或者配置列表时特别有用。一些实际用起来的建议给枚举成员起名字时尽量用全大写这是常量的惯例。名字要能清晰表达含义别用缩写除非那个缩写是行业共识。如果枚举值需要持久化比如存数据库建议存整数值而不是名字。因为名字可能会变但数字值相对稳定。从数据库读出来时再用枚举类包装一下。在函数签名里可以考虑用枚举类型作为参数类型提示。虽然不是强制约束但能给使用者和IDE清晰的提示。defset_status(status:Status)-None:# 函数体别滥用继承。IntEnum之间最好不要有继承关系因为枚举的本意就是一组有限的值继承容易让逻辑变得复杂。如果枚举值需要跨模块共享考虑单独放在一个constants.py或者enums.py文件里集中管理。和其他方式的对比和普通Enum比IntEnum最大的优势就是可以直接当整数用。如果你需要枚举值参与数学运算、当数组索引、或者和其他整数API交互IntEnum是更自然的选择。普通Enum虽然也能通过.value获取值但多了一层间接访问。和直接用类变量定义常量相比IntEnum提供了更好的封装和约束。类变量可以被随意修改而枚举成员是只读的。枚举还提供了迭代、按值查找等额外功能。和字典相比IntEnum有类型信息IDE能提供自动补全调用时不容易拼错键名。性能上也稍好一些因为属性访问比字典查找快。不过IntEnum也不是万能的。如果你的“常量”需要是字符串或者其他非整数类型那就得用普通的Enum。如果值的集合是动态的、运行时可能变化的那可能还是得用字典或者别的数据结构。说到底IntEnum是给那些有限的、已知的、需要名字的整数常量准备的。它让代码更清晰维护更简单算是一个小投入大回报的工具。下次在代码里写数字常量时不妨想想能不能用IntEnum包装一下给它们一个体面的身份。