Python 3.8下如何为含枚举键的字典做类型提示？

问题原因 #

TypedDict的定义要求字段名必须是字符串字面量，不能使用变量（比如MyEnum.A.value），这是导致mypy报错的核心原因。

符合Python风格的解决方案 #

方案一：直接使用枚举值的字面量定义TypedDict

这是最直观且符合Python"显式优于隐式"风格的方案。既然枚举值是固定的字符串，直接将这些字符串字面量作为TypedDict的字段名，既能通过mypy检查，又能精确指定每个键对应的类型。

from typing import TypedDict
from enum import Enum

class MyEnum(Enum):
    A = 'a'
    B = 'b'

class MyClass(TypedDict):
    """定义函数返回的字典结构"""
    'a': int
    'b': float

def foo() -> MyClass:
    res: MyClass = {MyEnum.A.value: 3, MyEnum.B.value: 4.4}
    return res

如果担心后续枚举值修改时TypedDict字段未同步，可以添加断言做静态检查：

# 确保所有枚举值都在TypedDict中有对应字段
assert set(item.value for item in MyEnum) == set(MyClass.__annotations__.keys())

方案二：用Literal结合Dict（适合类型精度要求较低的场景）

如果不需要区分每个键的具体类型，仅需约束键为枚举值集合，可以使用Literal和Dict的组合，但这种方式无法实现每个键对应不同类型的精确检查：

from typing import Dict, Literal, Union
from enum import Enum

class MyEnum(Enum):
    A = 'a'
    B = 'b'

def foo() -> Dict[Literal['a', 'b'], Union[int, float]]:
    return {MyEnum.A.value: 3, MyEnum.B.value: 4.4}

总结 #

优先选择方案一，它既能满足精确的类型检查需求，又保持了代码的清晰性和可维护性，完全符合Python的设计风格。

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