在Python的类型提示中，如果你想判断一个UnionType是否包含特定的类型，可以使用typing模块中的get_origin和get_args函数来实现。以下是一个示例代码，展示如何进行这种判断：fromtypingimportUnion,get_origin,get_argsdefcheck_union_contains_type(union_type,target_type):origi

本文详解如何在Python类型提示中判断UnionType是否包含特定类型。 面对UnionType（如`str | None`）无法直接使用`in`操作符的难题，文章提供了两种解决方案：一是利用`typing.get_args()`函数获取UnionType的子类型，再用`in`操作符判断目标类型是否存在；二是使用`isinstance()`函数，通过创建目标类型实例间接判断兼容性。 两种方法都避免了直接访问UnionType成员的限制，有效解决了类型检查问题，最终实现对函数参数类型提示中特定类型的精准判断，提升代码可靠性。

Python类型提示中的UnionType成员访问详解

Python类型提示中，UnionType表示一个变量可以是多种类型中的一种。本文探讨如何有效判断UnionType是否包含特定类型，尤其在函数参数类型提示的场景下。

问题： 我们需要检查函数参数类型提示中是否包含特定类型，例如str。对于简单类型（如int）直接判断即可，但UnionType（例如str | None）并非可迭代对象，无法直接用in操作符判断。

例如，以下代码尝试检查get_score函数的name参数类型提示是否包含str：

from typing import Union, Callable
from types import UnionType
from inspect import signature
from loguru import logger

def check_func_args_hints(func: Callable) -> bool:
    typed_signature = signature(func).parameters.items()
    for parameter_name, parameter in typed_signature:
        if parameter_name != 'name':
            continue
        logger.debug(parameter.annotation)
        logger.debug(type(parameter.annotation))
        logger.debug(f'----------------')

def get_score(name: str | None = None) -> float | None:
    pass

check_func_args_hints(get_score)

输出显示parameter.annotation为types.UnionType，无法直接判断是否包含str。

解决方案：

方法一：使用typing.get_args()

typing.get_args()函数可以获取UnionType的子类型。我们可以用它来判断UnionType中是否包含str：

from typing import Union, Callable, get_args
from inspect import signature

def check_func_args_hints(func: Callable) -> bool:
    typed_signature = signature(func).parameters.items()
    for parameter_name, parameter in typed_signature:
        if parameter_name == 'name':
            if isinstance(parameter.annotation, UnionType):
                return str in get_args(parameter.annotation)
            else:
                return parameter.annotation is str
    return False

def get_score(name: str | None = None) -> float | None:
    pass

print(check_func_args_hints(get_score))  # 输出 True

get_args()返回一个包含UnionType所有子类型的元组，可以直接使用in操作符判断。如果不是UnionType，则直接判断是否为str类型。

方法二：使用isinstance()

无需直接访问UnionType成员，可以使用isinstance()函数达到相同目的。创建一个目标类型的实例，并用isinstance()检查它是否与parameter.annotation兼容：

from typing import Union, Callable
from inspect import signature

def check_func_args_hints(func: Callable) -> bool:
    typed_signature = signature(func).parameters.items()
    for parameter_name, parameter in typed_signature:
        if parameter_name == 'name':
            return isinstance("", parameter.annotation)  # 使用空字符串作为str实例
    return False

def get_score(name: str | None = None) -> float | None:
    pass

print(check_func_args_hints(get_score))  # 输出 True

isinstance()会检查对象是否属于指定类型或其子类型，间接判断UnionType中是否包含目标类型。 这里用空字符串""作为str实例，任何str实例都可行。

两种方法都能有效解决UnionType不可迭代的问题，准确判断目标类型是否存在于类型提示中。 选择哪种方法取决于个人偏好和代码风格。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《在Python的类型提示中，如果你想判断一个UnionType是否包含特定的类型，可以使用typing模块中的get_origin和get_args函数来实现。以下是一个示例代码，展示如何进行这种判断：fromtypingimportUnion,get_origin,get_argsdefcheck_union_contains_type(union_type,target_type):origin=get_origin(union_type)iforiginisUnion:args=get_args(union_type)returntarget_typeinargsreturnFalse#示例使用UnionType=Union[int,str,float]print(check_union_contains_type(UnionType,int))#输出:Trueprint(check_union_contains_type(UnionType,list))#输出:False在这个代码中：get_origin函数用于获取类型的原始类型。对于Union[int,str,float]，get_origin会返回Union。get_args函数用于获取Union中的所有类型参数。对于Union[int,str,float]，get_args会返回一个包含int、str和float的元组。然后，我们检查target_type是否在args中，如果是，则返回True，否则返回False。这种方法可以帮助你判断一个UnionType是否包含特定的类型。》文章吧，也可关注golang学习网公众号了解相关技术文章。