Python泛型函数类型注解实现：让Factory函数返回传入类型的具体实例

你现在的问题核心是要让工厂函数的类型注解能精准关联传入的类和返回的实例类型，而不是模糊的父类或者Union类型。Python标准库typing里的TypeVar和Type工具正好能搞定这个需求，具体做法如下：

1. 导入必要的类型工具 #

首先从typing模块导入我们需要的泛型相关工具（Python 3.9+也可以用collections.abc，不过用typing兼容性更好）：

from typing import TypeVar, Type

# 先定义你的基类示例
class BaseCustomObject:
    def __init__(self, param: int):
        self.param = param

2. 定义绑定到基类的泛型变量 #

我们需要创建一个泛型变量T，并把它绑定到BaseCustomObject——这样可以限制传入工厂函数的类必须是这个基类的子类，避免传入无关类型：

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

3. 修正工厂函数的类型注解 #

现在把工厂函数的参数和返回值用泛型类型重新注解：

def calculation(param: int) -> int:
    # 这里是你的计算逻辑示例
    return param * 2

def factory(object_to_create: Type[T], parameter: int = 1) -> T:
    constructor_parameter = calculation(parameter)
    return object_to_create(constructor_parameter)

4. 测试使用场景 #

当你传入具体的子类时，IDE和类型检查工具（比如mypy）会自动精准推断返回的实例类型：

class MyCustomObject(BaseCustomObject):
    def my_unique_method(self):
        print("This is a method only for MyCustomObject")

# IDE会自动识别my_custom_object_instance的类型是MyCustomObject
my_custom_object_instance = factory(MyCustomObject)
# 此时调用子类独有的方法不会有类型提示错误
my_custom_object_instance.my_unique_method()

为什么这能解决你的问题？ #

• Type[T]明确表示传入的参数是T类型的类本身（而不是T的实例）
• 返回值T直接关联到传入的类，确保返回的是该类的实例
• bound=BaseCustomObject做了类型约束，保证传入的类都是BaseCustomObject的子类，符合你的业务逻辑

这样不管你传入哪个BaseCustomObject的子类，类型检查工具和IDE都能准确识别返回的具体类型，完全不需要用Union类型就能满足你的需求。

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