将Python联合类型参数传入非联合类型函数的类型检查问题

你的问题核心是mypy无法推断x["c"]的具体类型——因为AttributeDict定义的是所有值都可能是int | str | list[str]，类型检查器无法确定某个键对应的值到底是哪种类型，所以会报错。以下是几种可行的解决方式：

1. 使用类型断言（Type Assertion） #

直接向mypy声明x["c"]的具体类型，适合你明确知道该键对应值类型的场景：

from typing import cast

AttributeDict = dict[str, int | str | list[str]]
x: AttributeDict = {"a": 1, "b": "2", "c": ["a", "b"]}
def my_func(arg1: list[str]) -> None:
    print(arg1)

# 用cast断言x["c"]是list[str]
my_func(cast(list[str], x["c"]))

注意：类型断言是开发者向类型检查器做出的承诺，运行时不会做类型验证，如果实际值不是list[str]，会引发运行时错误。

2. 改用TypedDict定义精确结构 #

如果清楚字典的所有键和对应类型，用TypedDict替代普通字典类型，能让mypy精确识别每个键的类型：

from typing import TypedDict

class MyAttributeDict(TypedDict):
    a: int
    b: str
    c: list[str]

x: MyAttributeDict = {"a": 1, "b": "2", "c": ["a", "b"]}
def my_func(arg1: list[str]) -> None:
    print(arg1)

# 此时mypy明确知道x["c"]是list[str]，不会报错
my_func(x["c"])

3. 运行时类型检查（类型窄化） #

通过isinstance做运行时类型判断，mypy会自动窄化变量类型，适合需要处理多种类型分支的场景：

AttributeDict = dict[str, int | str | list[str]]
x: AttributeDict = {"a": 1, "b": "2", "c": ["a", "b"]}
def my_func(arg1: list[str]) -> None:
    print(arg1)

val = x["c"]
if isinstance(val, list):
    # mypy能推断这里val是list[str]
    my_func(val)
else:
    # 处理非列表类型的情况，比如抛出异常或跳过
    raise ValueError("Expected list for key 'c'")

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