Python 标准：返回共享基类的多个返回类型的类型提示

如何解决Python 标准：返回共享基类的多个返回类型的类型提示

关于类型提示返回多个继承类的函数的快速问题。

class Sub:
    pass

class A(Sub):
    pass

class B(Sub):
    pass

import typing

def func() -> typing.Union[A,B]:
    if cond1:
        return A
    if cond2:
        return B

#or should you do this?

def func() -> Sub:
    if cond1:
        return A
    if cond2:
        return B

返回子类的类型提示更合适还是应该使用typing.Union

返回所有这些。

解决方法

Mypy 是一种流行的 Python 静态类型检查器，它提到可以使用具有上限的类型变量来处理与您提到的情况类似的情况：

• The type of class objects 描述了如何使用 Type 和具有上限的类型变量来注释接收类并返回该类实例的函数：

U = TypeVar('U',bound=User)

def new_user(user_class: Type[U]) -> U:
    # Same  implementation as before

这种方法的好处似乎是检查器不会拒绝对仅存在于子类中而不存在于超类中的方法的调用：

def new_user(user_class: type) -> User:
    # Same  implementation as before

buyer = new_user(ProUser)
buyer.pay()  # Rejected,not a method on User

• Type variables with upper bounds 描述了如何创建和使用具有上限的类型变量：

类型变量也可以限制为具有特定类型的子类型的值。此类型称为类型变量的上限，并使用 bound=... 的 TypeVar 关键字参数指定。

在你的情况下，你可以使用这个：

from typing import TypeVar

S = TypeVar("S",bound=Sub)

def func() -> S:
    # Your implementation