覆蓋 __init__ 中參數的類型提示

Question

在此代碼上：

class Obj():
    def a(self):
        pass


class ObjChild(Obj):
    def b(self):
        pass


class A:
    def __init__(self, obj: Obj):
        self.obj = obj


class B(A):
    def __init__(self, obj: ObjChild):
        super().__init__(obj)
        
    def kuku(self):
        self.obj.b()

Pycharm 警告我最后一行：

class 'Obj' 的未解析屬性參考 'b'

但是我將在__init__中傳遞的obj類型提示為ObjChild ，因此它應該靜態地知道self.obj是一個ObjChild實例，並且具有b方法。

我怎樣才能讓 Pycharm 意識到這一點？ 它弄亂了我在項目中的自動完成功能。

Answer 1

使A Generic ：

from typing import Generic, TypeVar


class Obj:
    def a(self):
        pass


_Obj = TypeVar('_Obj', bound=Obj)


class ObjChild(Obj):
    def b(self):
        pass


class A(Generic[_Obj]):
    def __init__(self, obj: _Obj):
        self.obj = obj


class B(A[ObjChild]):
    def __init__(self, obj: ObjChild):
        super().__init__(obj)
        self.obj.b()

除非您將A設為泛型，否則將ObjChild傳遞給A.__init__和A.obj是ObjChild之間沒有任何聯系。