如何让通过API创建的Python Enum枚举类支持vscode type hinting类型提示？

通过Enum API动态创建的枚举属于运行时生成结构，VS Code使用的Pylance等静态类型检查器无法在不执行代码的情况下推断动态传入的字典键值，因此默认无法生成类型提示，可通过以下方案解决：

• 方案1：使用TYPE_CHECKING分支双端适配
  该方案完全不影响运行时逻辑，同时让编辑器能完整识别枚举成员，是适用性最高的解法，示例代码如下：

from enum import Enum
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    # 静态类型检查阶段会读取该分支的定义，支持枚举成员的点选、类型校验
    class Color(Enum):
        RED = 1
        YELLOW = 2
        GREEN = 3
else:
    # 运行时执行原有的动态创建逻辑
    color_values = dict(RED = 1, YELLOW = 2, GREEN = 3)
    Color = Enum('Color', color_values, type=int)

• 方案2：编写独立存根文件（.pyi）
  如果存在批量动态生成的枚举，或不想在业务代码中混入静态定义，可以在代码文件同级目录下创建同名.pyi存根文件，类型检查器会优先读取存根中的类型定义。
  例如业务代码存放在color.py中，就新建color.pyi，内容如下：

from enum import Enum

class Color(Enum):
    RED: int
    YELLOW: int
    GREEN: int

color.py中保留原有动态创建的代码即可，编辑器会自动匹配存根的类型声明。

• 方案3：显式标注枚举类型
  如果仅需要校验枚举值的合法范围，不需要成员点选提示，可以用TypeAlias+Literal组合完成类型标注：

from enum import Enum
from typing import Literal, TypeAlias

ColorValue: TypeAlias = Literal[1, 2, 3]
Color: Enum[ColorValue]

color_values = dict(RED = 1, YELLOW = 2, GREEN = 3)
Color = Enum('Color', color_values, type=int)

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