Python类装饰后Pylance无代码提示问题的排查与解决咨询

问题根源 #

你的装饰器类型注解存在关键错误：装饰器接收的是类对象（比如User这个类本身），但你把参数类型标注成了PrimaryModel——这是类的实例类型，而非类类型。

Pylance会严格遵循类型注解做推导，它会错误认为mydeco返回的是PrimaryModel实例，而非类本身，进而引发一系列类型识别混乱：

• 直接实例化被装饰的User时，Pylance无法识别User()返回的是User实例，只能当成Any类型，因此没有代码提示。
• 继承被装饰的User时，Pylance会误以为User是一个PrimaryModel实例（而非类），子类Player的继承关系完全不符合类型系统逻辑，导致后续哪怕显式声明类型，也无法修复推导混乱。

解决方案 #

1. 修正装饰器的类型注解 #

使用Type[PrimaryModel]（Python 3.8及以下需从typing导入Type，Python 3.9+可直接用type[PrimaryModel]）表示“PrimaryModel的类类型”：

from typing import Type

class PrimaryModel():
    def print(self):
        print("I am invisible O.o")

# 注解修正：接收PrimaryModel类对象，返回同类型的类
def mydeco(cls: Type[PrimaryModel]) -> Type[PrimaryModel]:
    return cls

if __name__ == "__main__":
    @mydeco
    class User(PrimaryModel):
        pass
    class Player(User):
        pass
    alan = Player()
    alan.print()  # 代码提示恢复正常

2. 泛型装饰器（通用方案） #

如果装饰器需要支持装饰任意类，而非仅PrimaryModel的子类，使用泛型可以让装饰器完全保留原类的类型信息，适配所有场景：

from typing import TypeVar, Type

# 定义泛型变量，代表任意类类型
T = TypeVar('T')

class PrimaryModel():
    def print(self):
        print("I am invisible O.o")

# 泛型装饰器：接收任意类，返回原类类型
def mydeco(cls: Type[T]) -> Type[T]:
    return cls

if __name__ == "__main__":
    @mydeco
    class User(PrimaryModel):
        def say_hello(self):
            print("Hello!")
    
    class Player(User):
        def run(self):
            print("Running!")
    
    alan = Player()
    alan.print()    # 有提示
    alan.say_hello()# 有提示
    alan.run()      # 有提示

为什么之前的临时方案有时失效 #

• 第一次显式声明alan: PrimaryModel有效，是因为直接强制指定了变量类型，绕过了装饰器导致的推导错误。
• 但继承被装饰的User时，Pylance已经因为错误的类型注解，将User识别为PrimaryModel实例而非类，子类Player的继承关系在类型系统中完全不成立，此时显式声明类型也无法修复底层的推导混乱。

内容的提问来源于stack exchange，提问作者Daniel Šebík