Pylance子类自动补全异常咨询：SubFoo类无法识别SubBar类的方法与属性

你的代码写法语法上是完全没问题的，但Pylance作为静态类型检查器，是严格基于类型标注来做自动补全和类型推断的。问题出在父类Foo的__init__方法里：你把self.attr显式标注成了Bar类型——哪怕你在SubFoo中传入的是SubBar实例，Pylance依然会遵循父类的类型定义，认定self.attr就是Bar类型，而Bar并没有method2方法，自然没法给你补全这个方法。

下面给你几个实用的修复方案，按推荐程度排序：

方案1：使用泛型（最优雅，优先推荐） #

让父类Foo支持泛型，这样它可以适配不同的Bar子类，子类SubFoo能明确指定自己使用的具体类型，Pylance就能准确识别属性的真实类型了。

修改后的代码如下：

from typing import Generic, TypeVar

# 定义一个类型变量，代表Bar及其所有子类
T = TypeVar('T', bound='Bar')

class Bar:
    def method1(self):
        # do stuff
        pass

class SubBar(Bar):
    def method2(self):
        # do stuff
        pass

class Foo(Generic[T]):
    def __init__(self, arg: T):
        self.attr: T = arg  # 用类型变量替代固定的Bar类型

    def do_a(self):
        self.attr.method1()

class SubFoo(Foo[SubBar]):
    def __init__(self, arg: SubBar):
        super().__init__(arg)

    def do_b(self):
        self.attr.method1()  # 正常补全
        self.attr.method2()  # 现在Pylance能识别并补全method2了

方案2：在子类中重新标注属性类型 #

如果你不想引入泛型，可以在SubFoo里重新给self.attr标注类型，直接覆盖父类的定义：

class Bar:
    def method1(self):
        # do stuff
        pass

class SubBar(Bar):
    def method2(self):
        # do stuff
        pass

class Foo:
    def __init__(self, arg: Bar):
        self.attr: Bar = arg

    def do_a(self):
        self.attr.method1()

class SubFoo(Foo):
    def __init__(self, arg: SubBar):
        super().__init__(arg)
        self.attr: SubBar = arg  # 重新标注为SubBar类型

    def do_b(self):
        self.attr.method1()
        self.attr.method2()  # 现在可以正常补全了

方案3：使用类型断言（临时应急用） #

如果只是在某个方法里临时需要识别SubBar的方法，可以用cast来告诉Pylance当前属性的实际类型：

from typing import cast

class Bar:
    def method1(self):
        # do stuff
        pass

class SubBar(Bar):
    def method2(self):
        # do stuff
        pass

class Foo:
    def __init__(self, arg: Bar):
        self.attr: Bar = arg

    def do_a(self):
        self.attr.method1()

class SubFoo(Foo):
    def __init__(self, arg: SubBar):
        super().__init__(arg)

    def do_b(self):
        self.attr.method1()
        # 用cast断言attr是SubBar类型
        sub_bar_attr = cast(SubBar, self.attr)
        sub_bar_attr.method2()  # 这里可以正常补全

这个方案虽然能解决问题，但每次使用都要写断言，代码不够简洁，只推荐作为临时解决方案。

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