如何修改Python EnumType以允许继承含成员的枚举类？

Python标准库的EnumMeta默认会阻止继承已包含成员的Enum类，核心原因是在类创建阶段会检查基类是否已定义枚举成员。以下是几种直接修改EnumMeta来移除该限制的可行方法：

1. 重写EnumMeta.__new__方法，移除基类成员检查 #

EnumMeta的__new__方法中存在一段逻辑：如果发现任何基类已经包含枚举成员（通过_enum_member_names_属性判断），就会抛出TypeError。我们可以自定义元类继承EnumMeta，并修改这段检查逻辑：

from enum import EnumMeta, Enum

class InheritableEnumMeta(EnumMeta):
    def __new__(metacls, cls, bases, classdict):
        try:
            # 尝试调用父类方法，若触发"无法扩展枚举"的错误则绕过原检查逻辑
            return super().__new__(metacls, cls, bases, classdict)
        except TypeError as e:
            if "cannot extend enumerations" in str(e):
                # 跳过EnumMeta的基类成员检查，直接调用元类的基础创建逻辑
                return super(EnumMeta, metacls).__new__(metacls, cls, bases, classdict)
            else:
                raise

# 基于自定义元类创建可继承的Enum基类
class InheritableEnum(Enum, metaclass=InheritableEnumMeta):
    pass

# 测试：创建带成员的基类Enum
class BaseEnum(InheritableEnum):
    VALUE1 = 1
    VALUE2 = 2

# 继承这个带成员的Enum
class SubEnum(BaseEnum):
    VALUE3 = 3
    VALUE4 = 4

# 验证成员是否被继承
print(list(SubEnum))  # 输出: [<BaseEnum.VALUE1: 1>, <BaseEnum.VALUE2: 2>, <SubEnum.VALUE3: 3>, <SubEnum.VALUE4: 4>]

2. 自定义元类实现成员合并逻辑 #

如果需要更严谨的处理（比如避免成员名冲突、支持子类覆盖基类成员），可以在自定义元类中主动合并基类和子类的成员：

from enum import EnumMeta, Enum

class MergeableEnumMeta(EnumMeta):
    def __new__(metacls, cls, bases, classdict):
        # 收集所有基类的枚举成员
        base_members = {}
        for base in bases:
            if isinstance(base, EnumMeta):
                base_members.update(base.__members__)
        
        # 合并子类成员，子类成员优先覆盖基类同名成员
        merged_classdict = {**base_members, **classdict}
        
        # 清理非枚举成员属性
        for attr in ['__module__', '__qualname__', '__doc__']:
            merged_classdict.pop(attr, None)
        
        # 调用父类方法创建枚举类
        return super().__new__(metacls, cls, bases, merged_classdict)

class MergeableEnum(Enum, metaclass=MergeableEnumMeta):
    pass

# 测试
class BaseEnum(MergeableEnum):
    VALUE1 = 1
    VALUE2 = 2

class SubEnum(BaseEnum):
    VALUE2 = 200  # 覆盖基类成员
    VALUE3 = 3

print(SubEnum.VALUE2.value)  # 输出: 200
print(list(SubEnum))  # 输出: [<BaseEnum.VALUE1: 1>, <SubEnum.VALUE2: 200>, <SubEnum.VALUE3: 3>]

注意事项 #

• 这种修改会打破Enum的默认设计约束，可能导致意外行为（比如成员名冲突、序列化/反序列化异常），建议仅在明确业务需求的场景下使用。
• 如果只是需要复用枚举成员，也可以考虑用组合而非继承的方式（比如在子类中直接引用基类成员），这种方式更符合Enum的设计规范。

内容的提问来源于stack exchange，提问作者Bradley bare