如何用Python类型提示表示受类泛型TypeVar约束的泛型方法？

要实现你展示的Java泛型逻辑，Python可以通过typing模块的TypeVar、Generic以及可选的Protocol/抽象类来完成，核心是让方法的泛型参数约束为类泛型参数的子类型。

完整实现代码 #

from typing import TypeVar, Generic, Protocol, Serializable
from abc import ABC, abstractmethod
from datetime import timedelta
from io import StringIO

# 定义类级泛型参数
T = TypeVar('T')
# 定义方法级泛型参数，约束为T的子类型（对应Java的<T_SUB extends T>）
T_SUB = TypeVar('T_SUB', bound=T)

# 方式1：用Protocol定义接口（无需抽象方法，更接近Java接口）
class Example(Generic[T], Protocol):
    def doSomething(self, input: T_SUB) -> T_SUB:
        ...

# 方式2：用抽象类定义接口（强制子类实现方法）
# class Example(Generic[T], ABC):
#     @abstractmethod
#     def doSomething(self, input: T_SUB) -> T_SUB:
#         ...

# 实现类，指定泛型参数为Serializable
class Impl(Example[Serializable]):
    def doSomething(self, input: T_SUB) -> T_SUB:
        return input

# 测试用法
if __name__ == "__main__":
    impl = Impl()
    
    # 传入StringIO（实现了Serializable），返回类型自动推导为StringIO
    str_buf_result = impl.doSomething(StringIO())
    # 传入timedelta（实现了Serializable），返回类型自动推导为timedelta
    delta_result = impl.doSomething(timedelta(minutes=1))
    
    # 如果传入非Serializable类型，mypy会报错
    # class NonSerializable:
    #     pass
    # impl.doSomething(NonSerializable())  # mypy错误提示：参数类型不兼容，需为Serializable的子类型

关键细节说明 #

• 泛型参数约束：T_SUB = TypeVar('T_SUB', bound=T) 对应Java中的<T_SUB extends T>，确保方法的参数类型必须是类泛型参数T的子类型。
• 接口定义：用Protocol可模拟Java的接口行为（无需实现方法），用ABC+abstractmethod则强制子类必须实现方法，根据需求选择即可。
• 类型检查：mypy会自动验证：
  • 传入doSomething的参数必须是Serializable的子类型（因为Impl继承了Example[Serializable]）
  • 返回值类型与输入参数类型完全一致，和Java示例的行为完全匹配

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