Python装饰器函数类型不匹配问题排查与解决求助

类型检查器的警告源于Callable参数的逆变特性：
虽然Class继承自BaseClass，但method1的参数b要求的是能接受Tuple[BaseClass, ...]的Callable，而example的参数是Tuple[Class, ...]。逆变规则规定：若一个函数期望处理父类型的集合，你不能传入仅能处理子类型集合的函数——后者无法兼容父类型实例，存在类型安全风险，因此类型检查器抛出警告。

以下三种方案可消除该警告，按需选择：

方案1：泛型统一类型约束 #

用TypeVar定义绑定到BaseClass的泛型类型，让method1和example的类型参数保持一致，兼容继承关系：

from typing import Type, Callable, Tuple, TypeVar

T = TypeVar('T', bound='BaseClass')

class BaseClass:
    pass

class Class(BaseClass):
    pass

def method1(
    a: Type[T],
    b: Callable[[Tuple[T, ...]], Callable[[Type], Type]],
    c: Tuple[T, ...],
) -> Callable[[Type], Type]:
    def decorator(cls: Type) -> Type:
        return cls
    return decorator

def example(*input: T) -> Callable[[Type], Type]:
    return method1(Class, example, input)

方案2：放宽example的参数类型 #

若业务允许example接受任意BaseClass子类实例，直接修改参数类型为BaseClass：

def example(*input: BaseClass) -> Callable[[Type], Type]:
    return method1(Class, example, input)

此方案让example的参数类型与method1的要求完全匹配，直接消除警告。

方案3：用Protocol实现结构类型匹配（Python 3.8+） #

通过Protocol定义兼容的调用签名，允许结构匹配而非严格的类型继承：

from typing import Protocol, Type, Callable, Tuple

class BaseClass:
    pass

class Class(BaseClass):
    pass

class MethodBCallable(Protocol):
    def __call__(self, *args: BaseClass) -> Callable[[Type], Type]:
        ...

def method1(
    a: Type[BaseClass],
    b: MethodBCallable,
    c: Tuple[BaseClass, ...],
) -> Callable[[Type], Type]:
    def decorator(cls: Type) -> Type:
        return cls
    return decorator

def example(*input: Class) -> Callable[[Type], Type]:
    return method1(Class, example, input)

Protocol会检查函数的调用结构是否兼容，只要example能接受BaseClass实例作为参数，类型检查器就会认可。

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