Python 3.8中如何为字符串值Enum添加精准类型提示？

步骤1：定义字符串Enum的基类 #

先创建一个同时继承str和Enum的基类，作为所有字符串类型Enum的统一父类：

from enum import Enum

class StrBasedEnum(str, Enum):
    pass

步骤2：改造现有字符串Enum #

让你的Stuff1继承这个基类，数值型Enum保持原有定义即可：

class Stuff1(StrBasedEnum):
    OPTION_A = "a"
    OPTION_B = "b"

class Stuff2(int, Enum):
    OPTION_1 = 1
    OPTION_2 = 2

步骤3：为encoder函数添加类型提示 #

直接用StrBasedEnum作为参数的类型注解，IDE会自动识别并拦截不符合要求的Enum传入：

def encoder(value: StrBasedEnum) -> str:
    # 可选：添加运行时类型检查，进一步确保动态场景下的类型安全
    if not isinstance(value, StrBasedEnum):
        raise TypeError("仅允许传入字符串类型的Enum")
    return str(value)

方案优势 #

• 完全适配Python 3.8，无需依赖3.11+的StrEnum特性
• 静态类型检查（如PyCharm、mypy）会直接提示传入Stuff2这类数值Enum的错误
• 不需要提前知晓Enum内部的具体枚举值，符合需求
• 可选的运行时检查可以在动态调用场景下避免类型错误

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