PyCharm无法识别装饰器修饰的类方法返回类型，如何解决？

问题描述 #

给Python类的方法添加装饰器时，遇到PyCharm类型识别的差异：

使用@语法装饰方法时，PyCharm无法识别方法的返回类型，自动补全功能失效：

from typing import Any, Callable, TypeVar
import functools  # 原代码遗漏该导入，需补充

T = TypeVar('T')


def my_decorator(func: Callable[..., T]) -> Callable[..., T]:
    @functools.wraps(func)
    def wrapped(*args: Any) -> T:
        print("decorator")
        print("Before the method is called")
        result = func(*args)
        print("After the method is called")
        return result

    return wrapped


class Foo:
    def __init__(self, param):
        super().__init__()
        self.param = param

    @my_decorator
    def bar(self, x: int) -> str:
        return str(self.param + x)

if __name__ == '__main__':
    a = Foo(2)
    res = a.bar(1)

而采用直接赋值的写法时，PyCharm能正确识别返回类型：

class Foo:
    def __init__(self, param):
        super().__init__()
        self.param = param

    def bar(self, x: int) -> str:
        return str(self.param + x)

    bar = my_decorator(bar)

两种写法的差异 #

从Python语法逻辑看，两种写法本质等价，都是把原方法bar替换为装饰器返回的包装函数。但PyCharm的类型检查器对两者的处理有细微区别：

• @语法是在类定义解析阶段直接将装饰器应用到方法对象上，此时PyCharm需要通过装饰器的类型注解推导最终方法的类型。如果装饰器用Callable[..., T]这种宽泛的泛型注解，PyCharm无法精准关联原方法的参数和返回类型。
• 直接赋值写法是先定义带完整类型注解的方法，再将装饰后的函数重新赋值给类属性。此时PyCharm可以先获取原方法的精确类型，再结合装饰器注解，更轻松地推导最终方法类型。

让PyCharm识别@语法的类型推导方法 #

优化装饰器的类型注解，用typing.ParamSpec（Python 3.10+）精确保留原函数的参数签名，就能让PyCharm正确识别@语法装饰后的方法类型：

from typing import Any, Callable, TypeVar, ParamSpec
import functools

P = ParamSpec('P')
T = TypeVar('T')


def my_decorator(func: Callable[P, T]) -> Callable[P, T]:
    @functools.wraps(func)
    def wrapped(*args: P.args, **kwargs: P.kwargs) -> T:
        print("decorator")
        print("Before the method is called")
        result = func(*args, **kwargs)
        print("After the method is called")
        return result

    return wrapped


class Foo:
    def __init__(self, param):
        super().__init__()
        self.param = param

    @my_decorator
    def bar(self, x: int) -> str:
        return str(self.param + x)

修改后，装饰器的类型注解能精准传递原函数的参数和返回类型信息，PyCharm即可恢复自动补全和返回类型提示功能。同时确保使用较新版本的PyCharm，旧版本对ParamSpec的支持可能不完善。

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