在python中具有继承的类型注释[重复]

Question

我正在尝试为使用特定基类的子类的类 init 方法提供类型提示。 Gooling 如何做到这一点告诉我使用 Type[BaseClass] 作为注释，但是检查一直告诉我我错了。

作为我的问题的一个例子，假设这是base.py：

class B:
    def __init__(self):
        return

..还有另一个class.py...

class SomeClass:

    def __init__(self, some_param: Type[B]):
        self.sp = some_param

现在，在第三个文件test.py 中以下列方式使用这些类给我带来了一些困惑：

from base import B
from class import SomeClass 

if __name__ == '__main__':
    sc = SomeClass(B())

我会猜到这是正确的，但 pycharm 检查用提示强调它：

预期类型“Type[B]”，改为“B”

所以由于这不起作用，我想我可能需要使用TypeVar，所以我将base.py 更改为：

class B:
    def __init__(self):
        return

TB = TypeVar('TB', bound=B)

和class.py 到：

class SomeClass:

    def __init__(self, some_param: Type[TB]):
        self.sp = some_param

但是，这只是将 pycharm 检查更改为：

预期类型“Type[TB]”，改为“B”

最后，如果我将class.py 设置为：

class SomeClass:

    TB = TypeVar('TB', bound=B)

    def __init__(self, some_param: Type[TB]):
        self.sp = some_param

test.py中的检查错误消失了，pycharm没有投诉！

（奇怪的是，如果我将if __name__ == '__main__': sc = SomeClass(B()) 移动到class.py，仍然会有投诉。）

正如我所说，我对如何在 Python 中真正使用基类打字感到有点困惑：

• 这一切都只是 pycharm 错误吗？

如果没有，

• 为什么Type[B] 不够/有效？
• 为什么需要在类中指定 TypeVar 而不能简单地导入 TypeVar？

Answer 1

Type[B] 表示参数应该是一个类对象本身。

例如，

class A:
    pass

class B:
    pass

class C(B):
    pass

def foo(x: Type[B]):
    pass

foo(A)    # fails
foo(B)    # passes
foo(C)    # passes
foo(B())  # fails - an instance of B is not B or a subclass of B

您只需要B 作为类型提示，以允许B（或B 的子类）的实例作为参数。

class SomeClass:

    def __init__(self, some_param: B):
        self.sp = some_param