时间戳

首页 > 解决方案 > python - 如何检查Union [...]中的任何参数在python中是否为None？

python - python - 如何检查Union [...]中的任何参数在python中是否为None？

问题描述

我想过滤 Union[..] 中的任何参数为无的代码。识别无参数的最佳方法是什么？

我知道参数的顺序在 Union[] 中并不重要（从输入 import Union 开始），但我如何进行常见的检查（可能是一个函数）来检测 None 参数。

from typing import Union
Union[T, None]

标签： pythonunion

解决方案

您可以使用函数属性__annotations__：

def my_func(a: Union[int, None], b: int, c: str):
    print(a,b,c)

print(my_func.__annotations__) # {'a': typing.Union[int, NoneType], 'b': <class 'int'>, 'c': <class 'str'>}

现在我们可以做一些事情以编程方式检查它：

from typing import _GenericAlias

def check_if_func_accepts_none(func):
    for key in func.__annotations__:
        if isinstance(func.__annotations__[key], type(None)):
            return True
        elif isinstance(func.__annotations__[key], _GenericAlias) and type(None) in func.__annotations__[key].__args__:
            return True
    return False

例子：

>>> def b(a:int, b:None):
...     print('hi')
...
>>> def c(x:Union[None,str], y:int):
...     print('hi')
...
>>> def d(z: int, s:str):
...     print('hi')
...
>>> check_if_func_accepts_none(b)
True
>>> check_if_func_accepts_none(c)
True
>>> check_if_func_accepts_none(d)
False
>>>

编辑：回答您的评论，Union直接检查对象：

type(None) in obj.__args__

True如果None存在则返回，False否则返回（假设obj是 a Union）

推荐阅读