如何在Python的IntEnum类体中运行自定义代码生成枚举成员？

Enum（包括IntEnum）的元类EnumMeta在处理类定义时，会把类体执行环境里的locals()替换成自定义的EnumDict。这个EnumDict有个严格限制：不允许重复设置同一个键。

你的代码里，for i in range(3)循环每次迭代都会更新循环变量i，相当于反复执行locals()['i'] = 当前值。第一次迭代i=0时，EnumDict记录了键'i'；第二次迭代要把i设为1时，EnumDict检测到'i'已经存在，直接抛出Attempted to reuse key: 'i'的错误。而普通类用的是标准字典，允许重复赋值，所以不会出问题。

方法1：使用Enum的函数式API #

这是最适配动态生成枚举成员的方式，和你预期的IntEnum('Test', [...])用法一致，也能兼容自定义__new__方法：

from enum import IntEnum

# 生成成员映射，实际场景可替换为复杂元组的处理逻辑
members = {chr(ord('A') + i): i for i in range(3)}
Test = IntEnum('Test', members)

# 如果需要自定义__new__，先定义基类
class CustomIntEnum(IntEnum):
    def __new__(cls, value):
        # 这里写你解析复杂元组的自定义逻辑
        obj = super().__new__(cls, value)
        # 示例：添加额外属性
        obj.extra_attr = value * 2
        return obj

# 用基类生成目标枚举
Test = CustomIntEnum('Test', members)

方法2：类体内部用局部函数封装逻辑 #

如果坚持在类体里写代码，可以用局部函数封装循环，避免循环变量污染类命名空间：

from enum import IntEnum

class Test(IntEnum):
    def _add_members():
        members = {}
        for i in range(3):
            members['ABC'[i]] = i
        # 将成员批量添加到类命名空间
        for name, value in members.items():
            vars()[name] = value
    _add_members()
    # 清理临时函数
    del _add_members

方法3：利用__init_subclass__钩子（Python 3.6+） #

通过子类初始化钩子动态添加成员，风格更简洁：

from enum import IntEnum

class Test(IntEnum):
    @classmethod
    def __init_subclass__(cls):
        for i in range(3):
            name = 'ABC'[i]
            setattr(cls, name, i)

内容的提问来源于stack exchange，提问作者Mad Physicist