如何为接收泛型类并返回特定实例的Python函数添加类型标注

要实现接收单个类型参数的泛型类、返回绑定指定Data类型的实例，你需要修正类型标注的方式——原代码里的Type[T]无法正确表示“泛型类本身”，得用带约束的TypeVar来定位泛型类的类型。

完整实现代码 #

from typing import TypeVar, Generic, Type

# 定义你的Data容器类型（示例）
class Data:
    pass

# 定义泛型类的类型变量，约束为带单个类型参数的泛型类
T = TypeVar('T')
GenericWithSingleParam = TypeVar('GenericWithSingleParam', bound=Generic[T])

# 你原有的泛型Collection类
class Collection(Generic[T]):
    def __init__(self):
        self.items: list[T] = []

# 目标函数的正确实现
def get_data(typ: Type[GenericWithSingleParam]) -> GenericWithSingleParam[Data]:
    return typ[Data]()

代码说明 #

1. GenericWithSingleParam类型变量：通过bound=Generic[T]约束，确保传入的typ必须是带有单个类型参数的泛型类（比如你的Collection）。
2. 类型标注修正：函数返回值GenericWithSingleParam[Data]明确表示，返回的是传入泛型类绑定Data类型后的实例。
3. 实例化逻辑：typ[Data]()就是把泛型类的类型参数替换为Data后创建实例，符合你的需求。

调用示例 #

# 传入Collection类，返回Collection[Data]实例
col = get_data(Collection)
# 此时col的类型会被类型检查器识别为Collection[Data]

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