[Python] @NotNull装饰器如何写比较好?

问题描述

Java中的 @NotNull, @Nullable (by jetbrains): https://www.jetbrains.com/hel...

Python中可以用装饰器实现, 但是目前没找到细粒度控制的: https://www.google.com/search...

希望提供一种细粒度控制的, 即: 具体到每一个参数的控制

只希望检查空指针即可, 类型限定不做

手动的, 繁琐的写法

def func(param1, param2, param3): # 明确表示, 不接受空指针(这里写法很繁琐, 是否有简化方法?) if not ((param1 is not None) and (param2 is not None) and (param3 is not None)):raise TypeError(’%s or %s or %s is None’ % (param1, param2, param3)) # 正常函数逻辑

装饰器的写法(未实现)

@not_null(param1, param2)def func(param1, param2, param3, *args, **kw):上下文环境

Python 3.5

问题解答

from functools import wrapsdef allow_jsonp(fun): @wraps(fun) def wrapper_fun(*args, **kwargs):for a in args: if not a:return ’args is null’return fun(*args, **kwargs) return wrapper_fun@allow_jsonpdef a(a): print (111111111)print (a(0))回答2：

from functools import wrapsdef not_none(func): @wraps(func) def wrapper(*args, **kwargs):if (None in args) or (None in kwargs.values()): raise ValueError(’Arguments must be not None’)else: return func(*args, **kwargs) return wrapper