类型提示中的子类

Question

我想允许使用Python 3进行类型提示，以接受某个类的子类。例如：

class A:
    pass

class B(A):
    pass

class C(A):
    pass

def process_any_subclass_type_of_A(cls: A):
    if cls == B:
        # do something
    elif cls == C:
        # do something else

现在，当输入以下代码时：

process_any_subclass_type_of_A(B)

我得到一个PyCharm IDE提示“期望的类型A，取而代之的是Type [B]。”

如何更改此处的类型提示以接受A的任何子类型？

据此（https://www.python.org/dev/peps/pep-0484/#type-definition-
syntax
，“该类型的参数也接受该类型是特定参数类型的子类型的表达式。”） ，我知道我的解决方案(cls: A)应该可行？

Answer 1

当您指定时cls: A，是说要cls使用type的
实例A。指定cls为类型A（或其子类型）使用的类对象的类型提示typing.Type。

from typing import Type
def process_any_subclass_type_of_A(cls: Type[A]):
    pass

From类对象的类型 ：

有时您想谈论从给定类继承的类对象。可以将其拼写为Type[C]哪里C是一堂课。换句话说，当C是类的名称时，使用C注释一个参数声明该参数是C（或的子类的C）实例，但是使用Type[C]注释作为变量的声明则声明该参数是源自C（的）类对象。或C本身）。