如何用字典映射枚举键定义Python类属性类型并通过Pylance检查？

问题场景 #

想要通过字典将枚举值映射到对应类型，以此定义Python类的属性类型，但常规写法会触发Pylance报错：Illegal type annotation: variable not allowed unless it is a type alias。示例代码如下：

from enum import Enum, auto

class Keys(Enum):
    Integer = auto()
    String = auto()

types = {Keys.Integer: int, Keys.String: str}

class MyClass:
    int_property: types[Keys.Integer] # 此处报错
    str_property: types[Keys.String]  # 此处报错

可行解决方案 #

1. 提取类型别名（最直接） #

把字典中映射的类型提前定义成类型别名，类属性直接使用别名即可，既保留字典映射关系，又能让Pylance正常识别类型：

from enum import Enum, auto

class Keys(Enum):
    Integer = auto()
    String = auto()

# 定义类型别名
IntegerType = int
StringType = str

types = {Keys.Integer: IntegerType, Keys.String: StringType}

class MyClass:
    int_property: IntegerType  # 正常识别类型，无报错
    str_property: StringType   # 正常识别类型，无报错

2. 显式声明类型别名（Python 3.10+） #

使用typing.TypeAlias显式标记类型别名，符合PEP规范，类型检查器识别更清晰：

from enum import Enum, auto
from typing import TypeAlias

class Keys(Enum):
    Integer = auto()
    String = auto()

# 显式声明类型别名
IntegerType: TypeAlias = int
StringType: TypeAlias = str

types = {Keys.Integer: IntegerType, Keys.String: StringType}

class MyClass:
    int_property: IntegerType
    str_property: StringType

3. 动态设置类注解 #

如果希望保留字典索引的写法逻辑，可以通过动态修改类的__annotations__属性来设置类型，避开静态检查对直接索引表达式的限制：

from enum import Enum, auto

class Keys(Enum):
    Integer = auto()
    String = auto()

types = {Keys.Integer: int, Keys.String: str}

class MyClass:
    pass

# 动态注入属性类型注解
MyClass.__annotations__["int_property"] = types[Keys.Integer]
MyClass.__annotations__["str_property"] = types[Keys.String]

报错原因说明 #

Pylance等静态类型检查器要求类型注解必须是静态可解析的类型表达式，而types[Keys.Integer]属于运行时才能计算的变量索引操作，不属于静态类型表达式范畴，因此会触发报错。只有被标记为类型别名的变量，才能在注解中直接使用。

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