Python类型提示：如何告诉X是Foo的子类？

Question

如何在Python中为类类型编写类型提示？ 请考虑以下代码：

class A(object):
    pass

class B(A):
    pass

def register(cls: type[A]):
    assert issubclass(cls, A)

 register(A)
 register(B)

type[A]是正确的写法吗？ 如果我只使用cls: A，这意味着cls是A的一个实例，但我想说cls是一个类/类型，它至少是A的子类。

具体来说，我想指出的是参数应该是 Django模型类型。

Answer 1

要解决一般情况，您必须使用合适的__subclasscheck__编写一个元类。可能，但麻烦。

在Django模型类的特定情况下，已经存在一个显式元类，因此注释该元类可以完成以下工作：

import django.db.model as model

def register(cls: model.base.ModelBase): ...

这将起作用，因为isinstance(models.Model, models.base.ModelBase)是真的。

Answer 2

其他当前（2016年9月22日）的答案似乎不正确。根据PEP 484（关于类型提示），存在一个类对象类型的提示，称为Type[C]。根据typing模块的文档，您可以使用typing.Type[C]来实现您想要的。我自己也在用Python 3.5.2。

引用the PEP：

Sometimes you want to talk about class objects, in particular class objects that inherit from a given class. This can be spelled as Type[C] where C is a class. To clarify: while C (when used as an annotation) refers to instances of class C , Type[C] refers to subclasses of C .

并引用the docs：

A variable annotated with C may accept a value of type C. In contrast, a variable annotated with Type[C] may accept values that are classes themselves – specifically, it will accept the class object of C.

并参考您的具体示例：

import typing

class A(object):
    pass

class B(A):
    pass

def register(cls: typing.Type[A]):
    assert issubclass(cls, A)

register(A)
register(B)

您可以使用mypy静态地检查此类代码，它应该在简单的情况下工作——但是要注意mypy是一个正在进行的工作，到目前为止，关于类型[C]提示还有几个问题有待解决。