泛型函数包装器类型参数被擦除问题咨询

结论 #

• 你的代码类型注解没有错误，这是Python类型系统的已知限制，不是mypy或pyright的bug。

原因解析 #

当前Python类型检查器无法自动保留泛型函数被装饰后的泛型参数信息，核心原因有两点：

1. 你定义的wrapper是针对单一具体类型T的函数设计的，它接收的Callable[[T], T]是绑定到具体类型的函数类型，而非泛型函数。当传入带泛型参数的fun2[T]时，类型检查器没法把原函数的泛型参数“提升”到装饰后函数的层级，只能推断出最宽泛的Never类型。
2. Python的类型系统（基于PEP 484及后续扩展）对高阶泛型装饰器的原生支持不足——装饰器需要显式声明自己能处理泛型函数，而不是只处理固定类型的函数。

修复方案 #

如果要让装饰器支持泛型函数，需要调整类型注解，让wrapper接收泛型函数类型而非单一类型的函数：

from collections.abc import Callable
from typing import TypeVar, Generic

# 定义绑定泛型函数的类型变量
F = TypeVar('F', bound=Callable[..., object])

class Delay(Generic[F]):
    def __init__(self, wrapped: F):
        self.wrapped = wrapped

def wrapper(wrapped: F) -> Callable[[], Delay[F]]:
    def wrapping() -> Delay[F]:
        return Delay(wrapped)
    return wrapping

修改后再装饰泛型函数，类型检查器就能正确保留泛型信息：

@wrapper
def fun2[T](arg: T) -> T:
    return arg

reveal_type(fun2) # mypy输出: "def () -> Delay[def [T] (T) -> T]"
reveal_type(fun2()) # mypy输出: "Delay[def [T] (T) -> T]"
reveal_type(fun2().wrapped("test")) # mypy输出: "builtins.str"

补充说明 #

这种限制源于Python泛型的标称式设计和类型变量绑定规则，装饰器作为高阶函数，若要传递泛型信息，必须显式使用绑定泛型函数的类型变量，不能仅针对单个类型参数做注解。

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