如何在Python Protocol中定义不可变的ClassVar类型

我明白你现在遇到的困境了——想用Protocol来统一约束那些带有不可变类级Enum变量的类，但试了好几种写法都卡壳了。其实问题的核心是你没在Protocol里正确标记这是类变量，同时结合不可变性的声明，我来给你一步步讲清楚：

首先，先看最直接的解决方案，完全适配你现有类结构的Protocol写法：

from typing import ClassVar, Protocol, Final
from enum import Enum

class MyEnum(Enum):
    A = 0
    B = 1

class MyProtocol(Protocol):
    # 用ClassVar标记这是类变量，Final标记它不可变
    type: ClassVar[Final[MyEnum]]

# 你的原有类完全不需要修改，类型检查器会自动识别它们符合Protocol
class MyClass1:
    type: Final = MyEnum.A

class MyClass2:
    type: Final = MyEnum.B

# 测试一下：类型检查器会正常通过这些调用
def process_protocol(obj: MyProtocol) -> None:
    pass

process_protocol(MyClass1)  # 没问题
process_protocol(MyClass2)  # 没问题

这个写法能生效的原因很简单：

• ClassVar告诉类型检查器（比如mypy、pyright），这个属性是属于类本身的，不是实例的属性；
• Final则和你在MyClass1、MyClass2里的声明对应，标记这个类变量是不可修改的；
• 类型检查器会完美匹配你的现有类，因为它们的结构完全符合Protocol的约束。

咱们回头看你之前的尝试为啥都不行：

• MyProtocol1用了@property：这是在声明实例属性的getter，但你的type是类变量，两者完全不是一个层级的属性，自然不匹配；
• MyProtocol2的type: Final[MyEnum]：没有加ClassVar，类型检查器会默认把它当成实例属性，和你的类变量不匹配；
• MyProtocol3和MyProtocol4的写法：Python本身就不允许在@property里嵌套ClassVar，也不支持@property+@classmethod的组合，所以这些写法从语法上就不合法。

如果你的需求更严格，想约束每个类的type必须是某个具体的Enum成员（比如MyClass1只能是MyEnum.A），还可以结合泛型和Literal来实现：

from typing import ClassVar, Protocol, Final, Generic, Literal, TypeVar

E = TypeVar('E', bound=MyEnum)

class MyGenericProtocol(Generic[E], Protocol):
    type: ClassVar[Final[E]]

class MyClass1:
    type: Final[Literal[MyEnum.A]] = MyEnum.A

class MyClass2:
    type: Final[Literal[MyEnum.B]] = MyEnum.B

# 只接受type为MyEnum.A的类
def process_only_a(obj: MyGenericProtocol[Literal[MyEnum.A]]) -> None:
    pass

process_only_a(MyClass1)  # 没问题
process_only_a(MyClass2)  # 类型检查器会报错，符合预期

这个泛型版本能给你更精细的类型约束，不过如果只是需要统一所有类都有不可变的类级Enum变量，第一种简单写法就足够用了。

内容来源于stack exchange