[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