Python类型提示中如何判断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是否包含特定类型？的详细内容，更多请关注知识资源分享宝库其它相关文章！