Python Enum子类化进阶：如何实现枚举整体验证（如成员数量校验）

能不能用__init__方法实现？ #

不行。因为__init__是每个枚举成员实例化时单独调用的方法，此时根本无法知道后续是否还有其他成员要创建。强行用静态变量计数的话，会出现多个枚举类互相干扰的问题（比如先定义一个合规枚举后，计数残留会影响下一个枚举的验证），完全不优雅且容易出bug。

推荐的两种实现方式 #

方式1：自定义元类（最贴合Enum的创建流程）

Enum的创建过程由EnumMeta元类控制，我们可以继承它并重写__new__方法，在整个枚举类创建完成后统一验证成员数量：

from enum import Enum, EnumMeta

class PreciselyTwoEnumMeta(EnumMeta):
    def __new__(metacls, cls_name, bases, class_dict):
        # 先让父类完成枚举类的创建
        enum_cls = super().__new__(metacls, cls_name, bases, class_dict)
        # 验证成员数量
        if len(enum_cls.__members__) != 2:
            raise ValueError(f"枚举类 {cls_name} 必须恰好包含2个成员")
        return enum_cls

# 基于自定义元类创建基类
class PreciselyTwoEnum(Enum, metaclass=PreciselyTwoEnumMeta):
    pass

# 测试用例
class Allowed(PreciselyTwoEnum):
    FOO = 1
    BAR = 2  # 正常运行

class Disallowed(PreciselyTwoEnum):
    BAZ = 3  # 触发 ValueError: 枚举类 Disallowed 必须恰好包含2个成员

方式2：使用__init_subclass__（更简洁）

Python 3.6+支持__init_subclass__钩子，它会在子类被定义时自动调用，此时可以直接访问子类的__members__属性获取所有枚举成员：

from enum import Enum

class PreciselyTwoEnum(Enum):
    def __init_subclass__(cls):
        super().__init_subclass__()
        member_count = len(cls.__members__)
        if member_count != 2:
            raise ValueError(f"枚举类 {cls.__name__} 必须有2个成员，当前有{member_count}个")

# 测试用例
class Allowed(PreciselyTwoEnum):
    FOO = 1
    BAR = 2  # 正常运行

class Disallowed(PreciselyTwoEnum):
    BAZ = 3  # 触发 ValueError: 枚举类 Disallowed 必须有2个成员，当前有1个

两种方式的对比 #

• 元类方式：完全贴合Enum的底层创建逻辑，适合需要深度定制枚举行为的场景。
• __init_subclass__方式：代码更简洁易懂，不需要理解元类的复杂逻辑，适合简单的子类验证需求。

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