泛型类型不变的协议联合引发类型检查报错，求非重构方案

问题原因 #

你的代码中，get_exporter返回的是Union[MyExporter[SampleA], MyExporter[SampleB]]，类型检查器无法关联同一个exporter实例的get_sample和process_sample的泛型类型——它只会认为sample是SampleA | SampleB，而process_sample分别需要SampleA或SampleB，因此触发类型错误。

无需架构改动的解决方案 #

方案1：使用函数重载（推荐）

通过为get_exporter添加重载签名，让类型检查器根据传入的name字面量推断出具体的泛型实例类型：

from typing import overload, Literal, TypeVar, Protocol, Union
from your_module import BaseSample, SampleA, SampleB, my_exporter_a, my_exporter_b

T = TypeVar("T", bound=BaseSample)

class MyExporter(Protocol[T]):
    def get_sample(self) -> T:
        ...

    def process_sample(self, sample: T) -> str:
        ...

# 添加重载签名
@overload
def get_exporter(name: Literal["a"]) -> MyExporter[SampleA]:
    ...

@overload
def get_exporter(name: Literal["b"]) -> MyExporter[SampleB]:
    ...

# 原实现保持不变
def get_exporter(name: str) -> Union[MyExporter[SampleA], MyExporter[SampleB]]:
    if name == "a":
        return my_exporter_a
    return my_exporter_b

# 调用时类型检查正常
exporter = get_exporter("a")
sample = exporter.get_sample()  # 类型推断为SampleA
output = exporter.process_sample(sample)  # 无类型错误

这种方式完全保留原有架构，只是添加了重载签名，类型检查器能精准识别get_exporter("a")返回的是MyExporter[SampleA]，从而正确推断sample的类型与process_sample的参数类型匹配。

方案2：使用类型强制转换（快速妥协）

如果不想添加重载，可以用cast手动指定exporter的具体类型，绕过类型检查器的限制：

from typing import cast

exporter = get_exporter("a")
sample = exporter.get_sample()
output = cast(MyExporter[SampleA], exporter).process_sample(sample)

注意：这种方式需要你确保传入的name和强制转换的类型一致，否则可能引入运行时错误，仅适合临时解决或明确类型场景。

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