Python UnionType类型提示中如何可靠地检查子成员类型？-Python教程-PHP中文网

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

DDD

发布： 2025-03-04 10:34:47

原创

366人浏览过

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类型。

立即学习“Python免费学习笔记（深入）”；

可灵大模型

可灵大模型（Kling）是由快手大模型团队自研打造的视频生成大模型

214

查看详情

关键在于使用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类型提示中如何可靠地检查子成员类型？的详细内容，更多请关注php中文网其它相关文章！