子类方法自动装饰后Pylance方法签名显示错误求助

问题分析 #

你的代码运行逻辑完全正确，但Pylance无法显示obj.method的真实签名，核心原因是装饰器缺少精确的类型注解——静态类型检查器无法推断出子类的method被重新赋值为装饰后的wrapper函数，只能识别子类中原始定义的method(self, arg1, arg2)签名。

解决方案：给装饰器添加类型注解 #

通过typing模块的ParamSpec、TypeVar和Callable，可以给装饰器添加精准的类型提示，让Pylance正确识别装饰后方法的真实签名。

完整代码示例 #

from typing import Callable, ParamSpec, TypeVar

# 定义参数规格和返回值类型变量，用于捕获原函数的参数和返回值信息
P = ParamSpec('P')
R = TypeVar('R')
Self = TypeVar('Self')

def decorator(func: Callable[P, R]) -> Callable[[Self, str, *P.args, **P.kwargs], R]:
    def wrapper(self: Self, arg0: str, *args: P.args, **kwargs: P.kwargs) -> R:
        print("Wrapper called")
        return func(self, *args, **kwargs)
    return wrapper

class Parent:
    def __init_subclass__(cls, *args, **kwargs):
        super().__init_subclass__(*args, **kwargs)
        # 自动为子类的method方法应用装饰器
        cls.method = decorator(cls.method)

    @decorator
    def method(self) -> None:
        pass

class Child(Parent):
    def method(self, arg1: str, arg2: str) -> None:
        print("Method called")

obj = Child()
obj.method("arg0_val", "arg1_val", "arg2_val")

关键说明 #

1. 类型变量的作用：
   • ParamSpec('P')：捕获被装饰函数的完整参数列表，保留原函数的参数结构
   • TypeVar('R')：捕获被装饰函数的返回值类型，保证类型一致性
   • TypeVar('Self')：表示类实例自身，确保装饰器能正确关联类方法的self参数
2. 装饰器的类型注解：明确指定装饰后的函数签名为(self, arg0, *原函数参数, **原函数关键字参数)，让Pylance能准确推断出obj.method的真实签名
3. __init_subclass__的作用：自动为所有子类的method方法应用装饰器，无需手动添加@decorator装饰器

效果验证 #

修改后，Pylance会显示obj.method的签名为：

method(self, arg0: str, arg1: str, arg2: str) -> None

完全匹配装饰器中wrapper的签名，同时代码运行逻辑保持正确。

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