使用 union 键入别名

Question

我目前有这种类型的别名，以及我的代码中的一些相关函数：

Constant = int

def operation(data: Union[Constant, OtherTypes]):
    if isinstance(data, Constant):
        # do something
    else:
        # do something else

现在，我想让Constant也代表另一种类型，比如浮点数。 这个常量别名在我的代码库中使用，所以我不想到处更改它。

我努力了：

Constant = (int, float)

这与isinstance配合得很好，但Union抱怨"TypeError: Union[arg, ...]: each arg must be a type."

然后我尝试过：

Constant = Union[int, float]

现在，问题来自isinstance ； 我得到"TypeError: Subscripted generics cannot be used with class and instance checks" 。

有没有办法做我想要实现的目标？

谢谢。

Answer 1

支持Union s 的isinstance附带 python 3.10。 因此，从该版本开始，第二个解决方案将起作用。

请参阅https://peps.python.org/pep-0604/ 。

Answer 2

对于 3.10 之前的 Python，如 Frank、MSeifert 和 Richard Xia 的 Python 3.6 答案中的 Check a variable against Union type at runtime 中所述，您可以执行以下操作：

Python 3.8+

使用typing.get_args(tp) function 来获取具有联合类型的元组，您可以在isinstance内部使用：

from typing import Union, get_args

Constant = Union[int, float]

def operation(data: Union[Constant, OtherTypes]):
    if isinstance(data, get_args(Constant)):
        # do something
    else:
        # do something else

get_args仅返回类型的 arguments 而不验证该类型是Union还是其他泛型类型，这似乎足以满足您的要求。

如果由于某种原因您还需要在运行时检查Constant类型是否专门为Union ，请使用typing.get_origin(tp) function：

from typing import Union, get_origin

if get_origin(Constant) is Union:
    # do something

Python 3.5.3+

在 3.8 之前， get_args和get_origin函数不存在，因此您需要使用未记录的属性__args__和__origin__来代替。

def operation(data: Union[Constant, OtherTypes]):
    if isinstance(data, Constant.__args__):
        # do something
    else:
        # do something else

这仍然适用于 3.10.5，但由于这些属性未记录在案，因此在任何未来的 Python 版本中，上面的代码片段可能会立即停止工作。

Python 3.5.0 至 3.5.2

类型提示在 Python 的 3.5.0 版本中实现。 在 3.5.2 之前，获取联合的 arguments 的属性名称是__union_params__ ：

def operation(data: Union[Constant, OtherTypes]):
    if isinstance(data, Constant.__union_params__):
        # do something
    else:
        # do something else

当然，这个属性只存在于Union类型中，所以如果需要检查一个类型是否为Union ，请检查该属性是否存在。

请注意，这仅适用于 Python 3.5.2，因为在 3.5.3 中他们将属性名称更改为__args__ 。