Python类型提示中的UnionType成员访问详解
Python类型提示中，UnionType表示一个变量可以是多种类型中的一种。本文探讨如何有效判断UnionType是否包含特定类型，尤其在函数参数类型提示的场景下。
问题： 我们需要检查函数参数类型提示中是否包含特定类型，例如str。对于简单类型（如int）直接判断即可，但UnionType（例如str | None）并非可迭代对象，无法直接用in操作符判断。
例如，以下代码尝试检查get_score函数的name参数类型提示是否包含str：
立即学习“Python免费学习笔记（深入）”；
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不可迭代的问题，准确判断目标类型是否存在于类型提示中。 选择哪种方法取决于个人偏好和代码风格。
