使用 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__ 。