如何为Python中的Enum枚举类添加正确的类型提示？

你原来的写法存在两个核心问题：

• 第一个参数options: Enum实际上要求传入枚举成员，而非枚举类本身，和需求完全相反
• 无法约束selected必须与options属于同一个枚举类，类型检查工具无法识别这种关联关系

正确的方式是使用泛型绑定枚举类和其成员的类型，确保类型一致性。具体实现如下：

from enum import Enum
from typing import TypeVar, Type, Optional

# 定义绑定到Enum的类型变量，关联枚举类和其成员
T = TypeVar('T', bound=Enum)

class Foo(Enum):
    ALPHA = 1
    BETA = 2
    GAMMA = 3

def switch(options: Type[T], selected: Optional[T] = None) -> T:
    # 示例逻辑：若selected是当前枚举的有效成员则返回，否则返回第一个成员
    if selected is not None and selected in options:
        return selected
    return next(iter(options))

写法说明： #

• Type[T]表示options参数必须是继承自Enum的类（比如Foo）
• Optional[T]表示selected为可选参数，若传入则必须是options对应枚举类的成员（比如Foo.ALPHA）
• 返回类型T保证返回值也是该枚举类的成员

这种写法能让mypy等类型检查工具正确识别类型约束，避免传入不匹配的枚举类和成员，同时保留代码灵活性，支持任意Enum子类。

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