Python type hinting如何实现接收类返回对应实例的带类型标注泛型方法

Python 标准库的 typing 模块提供的 TypeVar 和 Type 就可以实现和 Java 等价的泛型类型标注效果，静态类型检查器（mypy、pyright 等）可以正确识别返回值的类型，完成类型校验和代码提示。

基础实现（无参构造场景） #

from typing import TypeVar, Type

# 定义泛型变量T
T = TypeVar('T')

def get_instance(clazz: Type[T]) -> T:
    # 这里可以根据需求加实例缓存、依赖注入等IoC逻辑
    return clazz()

使用示例 #

# 示例业务类
class OrderService:
    def create_order(self, user_id: int, amount: float) -> str:
        return f"order_{user_id}_{amount}"

# 调用方法
order_service = get_instance(OrderService)
# 静态检查器会自动识别order_service是OrderService类型，调用方法时会有提示，传参错误会抛出类型错误
order_service.create_order(123, 99.9)

扩展场景 #

1. 支持带构造参数的类 #

如果需要给实例构造传参，调整方法签名即可，静态检查器会同步校验构造参数的合法性：

from typing import TypeVar, Type

T = TypeVar('T')

def get_instance(clazz: Type[T], *args, **kwargs) -> T:
    return clazz(*args, **kwargs)

# 带构造参数的类示例
class UserService:
    def __init__(self, db_host: str, db_port: int):
        self.db_host = db_host
        self.db_port = db_port

# 传参不符合构造要求时静态检查器会直接报错
user_service = get_instance(UserService, "127.0.0.1", 3306)

2. IoC容器单例缓存场景 #

如果是IoC容器常用的单例模式，只需要调整内部实现，类型标注保持不变即可：

from typing import TypeVar, Type, cast

T = TypeVar('T')
_instance_cache: dict[Type, object] = {}

def get_instance(clazz: Type[T]) -> T:
    if clazz not in _instance_cache:
        _instance_cache[clazz] = clazz()
    return cast(T, _instance_cache[clazz])

版本适配说明 #

• Python 3.9+ 可以直接用内置的 type[T] 代替 typing.Type[T]，不需要额外导入Type
• 所有标注均符合PEP 484类型规范，主流IDE（PyCharm、VS Code）和静态检查工具都可以完美支持

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