Python中如何定义字符串Union作为字典键解决Pyright类型检查报错

你之前的写法中，Union["fruits", "cars", "vegetables"] 里的字符串被引号包裹后，会被类型检查器识别为前向引用声明，即你声明了一个名为fruits、暂未定义的类型，自然会抛出「未定义」错误。你实际需要的是「值固定为对应字符串的字面量类型」，应该用Literal实现。

Literal是Python 3.8才加入标准库typing模块的类型，3.7版本需要先安装第三方兼容包：
pip install typing-extensions

然后调整代码如下：

from typing import Optional, Dict
from typing_extensions import Literal

# 直接声明允许的字符串字面量，不需要额外套Union
PossibleKey = Literal["fruits", "cars", "vegetables"]
PossibleType = Dict[PossibleKey, str]

def some_function(target: Optional[PossibleType] = None):
  if target:
    all_fruits = target["fruits"]
    print(f"I have {all_fruits}")

• 完全符合Python类型注解规范，可正常通过Pyright/Pylance检查
• 保留了字典键的静态校验能力，访问不存在的键时类型检查器会提前报错，还能触发IDE的键名自动补全
• 类型本质还是字符串，和你现有处理Dict[str, Any]的逻辑完全兼容，不会出现Enum方案需要转枚举值的额外适配成本

如果后续升级到Python 3.8及以上版本，只需把导入语句改为from typing import Literal即可，无需修改其他代码。

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