[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
問題解答
回答1: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
相關文章:
