Python UnionType类型提示中如何可靠地检查子成员类型？

本文深入探讨Python中UnionType类型提示的可靠子成员类型检查方法。在静态代码分析和类型检查中，准确判断函数参数类型提示（尤其包含UnionType，例如`int | str`）是否包含特定类型至关重要。文章详细介绍了如何利用`typing.get_args`函数获取UnionType的子类型元组，从而可靠地判断其是否包含目标类型，例如`str`。相比之下，直接使用`isinstance`方法虽然简洁，但可靠性较低。文中通过代码示例和详细解释，清晰地阐述了`typing.get_args`方法的优势，并提供了完整的代码实现，帮助开发者有效解决Python UnionType类型检查难题。

深入解析Python UnionType子成员类型检查

本文探讨如何在Python中可靠地检查函数参数类型提示中是否包含特定类型，尤其是在使用UnionType时。这对于静态代码分析和类型检查至关重要。

直接检查简单类型（如int、str）比较容易，但UnionType（例如int | str）则需要更细致的处理。UnionType表示参数可接受多种类型，简单的in操作符或迭代无法直接判断其是否包含特定类型。

例如，函数get_score的name参数类型提示为str | None，inspect.signature返回的annotation为types.UnionType。我们需要确定该UnionType是否包含str类型。

关键在于使用typing.get_args函数。它能获取泛型类型的参数。对于UnionType，它返回一个包含所有子类型的元组。因此，我们可以这样判断UnionType是否包含str类型：

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

def check_func_args_hints(func: Callable) -> bool:
    for param in signature(func).parameters.values():
        if param.name != 'name':
            continue

        logger.debug(f"Annotation: {param.annotation}")
        logger.debug(f"Annotation Type: {type(param.annotation)}")
        if isinstance(param.annotation, UnionType):
            if str in get_args(param.annotation):
                logger.info("name parameter supports str")
                return True
            else:
                logger.info("name parameter does not support str")
                return False
        elif param.annotation is str:  # 使用 is 而不是 == 进行类型比较
            logger.info("name parameter supports str")
            return True
        else:
            logger.info("name parameter does not support str")
            return False

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

check_func_args_hints(get_score)

def another_func(name: int):
    pass

check_func_args_hints(another_func)

代码首先使用signature获取函数参数信息，然后检查参数名是否为name。如果是，则检查annotation的类型。如果是UnionType，则使用get_args获取子类型并判断是否包含str。否则，直接判断annotation是否为str。（代码已优化，使用 is 替代 == 进行类型比较，更准确）

另一种方法是使用isinstance：

isinstance("", param.annotation)

这种方法简洁，但可靠性取决于UnionType的定义和isinstance的实现，可能不如typing.get_args可靠。因此，typing.get_args是更推荐的方法。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Python UnionType类型提示中如何可靠地检查子成员类型？》文章吧，也可关注golang学习网公众号了解相关技术文章。