Python泛型基类中如何参数化@classmethod的cls类型以支持子类的正确类型推导

这个问题确实是Python泛型继承场景里的典型痛点——我们希望基类的bind方法能返回子类的泛型实例，而不是退化成基类实例，同时还能通过类型检查。我来一步步给你梳理正确的解决思路：

问题根源回顾 #

原来的实现中，unit方法直接返回From[U](val)，这就导致子类（比如Just）调用bind时，最终返回的是From[U]而不是Just[U]。如果改成直接用cls(val)，类型检查器又会报错，因为此时cls已经被参数化为From[T]，无法接受U类型的值。

正确的实现方案：用绑定的TypeVar注解cls类型 #

我们需要告诉类型检查器：cls不仅是From类，还可以是它的子类，并且这个类本身是泛型的，可以接受新的类型参数U。具体步骤如下：

1. 定义一个额外的TypeVar，绑定到From泛型类本身，这样cls可以是From或它的任何子类。
2. 给unit方法的cls参数加上正确的类型注解，让类型检查器知道cls能接受新的类型参数U，并返回对应子类的泛型实例。

完整代码示例：

from typing import Callable, TypeVar, Generic, Type, Any

T = TypeVar("T")
U = TypeVar("U")
# 定义一个TypeVar，绑定到From[Any]，表示它可以是From或任何子类的泛型类型
FromSubclass = TypeVar("FromSubclass", bound="From[Any]")

class From(Generic[T]):
    def __init__(self, val: T):
        self.val = val
    
    @classmethod
    def unit(cls: Type[FromSubclass], val: U) -> FromSubclass[U]:
        # 此时cls可以是From或它的子类，并且能接受U作为新的类型参数
        return cls(val)
    
    def bind(self, func: Callable[[T], U]) -> "From[U]":
        return self.unit(func(self.val))

# 子类继承From
class Just(From[T]):
    pass

# 测试代码
def to_string(i: int) -> str: 
    return str(i)

# 此时类型检查器会正确识别：返回的是Just[str]，而不是From[str]
result = Just(1).bind(to_string)
print(type(result))  # <class '__main__.Just'>
print(result.val)    # "1"

兼容Python 3.11+的简化方案：用Self泛型 #

如果你使用的是Python 3.11及以上版本，可以借助typing.Self（旧版本可使用typing_extensions.Self）来更简洁地实现，Self能自动关联当前类的泛型参数：

from typing import Callable, TypeVar, Generic, Self

T = TypeVar("T")
U = TypeVar("U")

class From(Generic[T]):
    def __init__(self, val: T):
        self.val = val
    
    @classmethod
    def unit(cls, val: U) -> Self[U]:
        return cls(val)
    
    def bind(self, func: Callable[[T], U]) -> Self[U]:
        return self.unit(func(self.val))

class Just(From[T]):
    pass

# 类型检查器会正确推导：返回值为Just[str]
Just(1).bind(to_string)

为什么原来的# type: ignore不是理想方案 #

虽然忽略类型检查能暂时绕过报错，但会丢失类型推导的核心优势——类型检查器无法再帮你发现后续代码中可能的类型错误（比如误将字符串当作整数操作result.val）。而上面的方案既能让子类正确返回自身的泛型实例，又能全程保持类型安全。

备注：内容来源于stack exchange，提问作者Seb