Python类型标注：如何约束函数参数为枚举类及其成员？

1. 为什么传入不同枚举的成员与类时MyPy不报错？ #

当你用TypeVar('T', bound=Enum)标注函数参数时，这个T只做宽泛的范畴约束：它仅要求传入的枚举类是Enum的子类，传入的成员是Enum的实例，但不会建立「成员必须属于传入的枚举类」的强关联。

MyPy只会校验参数是否落在Enum的范畴内，不会验证两者是否属于同一个具体的枚举类型，所以就算你把MyEnumA的成员和MyEnumB的类一起传入，它也会判定类型合法，不会触发报错。

2. 如何正确标注类型，让MyPy检测不匹配错误？ #

要让MyPy能关联枚举类和它的成员，需要用泛型绑定让两者的类型严格对应。具体步骤：

• 定义一个绑定到Enum的TypeVar
• 把枚举类参数的类型设为Type[EnumType]（表示EnumType这个枚举类型的类对象）
• 把成员参数的类型设为EnumType（表示EnumType类型的实例，也就是该枚举类的成员）

这样MyPy就能明确校验：成员必须是对应枚举类的实例，不匹配时直接触发报错。

示例代码 #

错误写法（无关联约束）

from enum import Enum
from typing import TypeVar, Type

T = TypeVar('T', bound=Enum)

def myfunction(enum_class: Type[T], enum_member: T) -> None:
    pass

class MyEnumA(Enum):
    VALUE_A = 1

class MyEnumB(Enum):
    VALUE_B = 2

# MyPy不会报错，但逻辑不合法
myfunction(MyEnumB, MyEnumA.VALUE_A)

正确写法（关联枚举类与成员）

from enum import Enum
from typing import TypeVar, Type

EnumType = TypeVar('EnumType', bound=Enum)

def myfunction(enum_class: Type[EnumType], enum_member: EnumType) -> None:
    # 可选：添加运行时校验，确保成员属于该枚举类
    assert enum_member in enum_class.__members__.values()

class MyEnumA(Enum):
    VALUE_A = 1

class MyEnumB(Enum):
    VALUE_B = 2

# 合法调用：成员与枚举类匹配，MyPy通过检查
myfunction(MyEnumA, MyEnumA.VALUE_A)
myfunction(MyEnumB, MyEnumB.VALUE_B)

# 不合法调用：成员与枚举类不匹配，MyPy触发报错
myfunction(MyEnumB, MyEnumA.VALUE_A)

MyPy报错信息（不合法调用时） #

error: Argument 2 to "myfunction" has incompatible type "MyEnumA"; expected "MyEnumB"  [arg-type]

这样就能精准检测出枚举类和成员不匹配的情况，符合预期的校验逻辑。

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