如何为匹配Protocol的枚举类型设置正确的类型标注？

问题根源 #

1. 原MyProto协议定义的是实例属性，但枚举的成员是类级属性，类型检查器无法匹配。
2. 函数check_item1接收的是枚举类本身（type[MyEnum]），但标注的是MyProto实例类型，类型不兼容。
3. 枚举成员的实际类型是枚举实例（如MyEnum.item1是MyEnum类型），虽然继承自int，但直接和int类型标注匹配会触发类型检查报错。

修正方案 #

需要调整协议定义和函数类型标注，让两者匹配枚举类的特性：

• 用ClassVar标记协议中的属性，明确这些是类级成员；
• 函数参数标注为type[MyProto]，表示接受符合协议的类类型；

修正后的代码 #

from typing import Protocol, ClassVar
from enum import Enum

class MyProto(Protocol):
    # 用ClassVar标记类级属性，匹配枚举成员的定义方式
    item1: ClassVar[int]
    item2: ClassVar[int]

# 参数e标注为type[MyProto]，接受符合协议的类类型
def check_item1(e: type[MyProto], value: int):
    return e.item1 == value

class MyEnum(int, Enum):
    item1 = 0
    item2 = 1
    other_item_not_in_protocol = 2

check_item1(MyEnum, 0)  # 类型检查通过

补充说明 #

• ClassVar用于标识类属性，告诉类型检查器这些属性属于类本身而非实例，完美适配枚举成员的特性；
• type[MyProto]表示“所有符合MyProto协议的类的类型”，这样传递MyEnum类时就能通过类型校验；
• 枚举成员虽然是MyEnum类型，但由于继承了int，在和int值比较时会自动解包为对应的整数值，类型检查器会认可这种兼容逻辑。

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