Python类型提示中的UnionType成员访问详解
Python类型提示中，UnionType表示一个变量可以是多种类型中的一种。本文探讨如何有效判断UnionType是否包含特定类型，尤其在函数参数类型提示的场景下。
问题： 我们需要检查函数参数类型提示中是否包含特定类型，例如str。对于简单类型（如int）直接判断即可，但UnionType（例如str | None）并非可迭代对象，无法直接用in操作符判断。
例如，以下代码尝试检查get_score函数的name参数类型提示是否包含str：
立即学习“Python免费学习笔记（深入）”；
from typing import Union, Callablefrom types import UnionTypefrom inspect import signaturefrom loguru import loggerdef 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: passcheck_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_argsfrom inspect import signaturedef 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 Falsedef get_score(name: str | None = None) -> float | None: passprint(check_func_args_hints(get_score)) # 输出 True
get_args()返回一个包含UnionType所有子类型的元组，可以直接使用in操作符判断。如果不是UnionType，则直接判断是否为str类型。
方法二：使用isinstance()
无需直接访问UnionType成员，可以使用isinstance()函数达到相同目的。创建一个目标类型的实例，并用isinstance()检查它是否与parameter.annotation兼容：
from typing import Union, Callablefrom inspect import signaturedef 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 Falsedef get_score(name: str | None = None) -> float | None: passprint(check_func_args_hints(get_score)) # 输出 True
isinstance()会检查对象是否属于指定类型或其子类型，间接判断UnionType中是否包含目标类型。 这里用空字符串””作为str实例，任何str实例都可行。
两种方法都能有效解决UnionType不可迭代的问题，准确判断目标类型是否存在于类型提示中。 选择哪种方法取决于个人偏好和代码风格。
以上就是Python类型提示中如何判断UnionType是否包含特定类型？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1358328.html
微信扫一扫 
支付宝扫一扫 
