python帶參裝飾器的改良版

?

簡單點就是這種

def deco2(param=1):def _deco2(fun):def __deco2(*args, **kwargs):print (param)fun(*args, **kwargs)return __deco2return _deco2
@deco2               # 錯誤的寫法
def f2(strx):print (strx)f2('hello')

?

運行上面這個，期待打印1和hello

實際上是不會打印任何東西的，因為@deco2后面忘了加括號，這樣就悲劇了，把f2這個函數名賦值給param了，所以運行f2函數，得到的結果是_deco2這個函數對象，并沒有去運行_deco2這個函數，所以沒有打印。

所以必須一定要改成@deco2(),你忘了寫括號就會不運行了，感覺蒙蔽，又要去找原因。

?

為了支持更簡潔的裝飾器，和減少失誤帶來的錯誤，比如忘了寫括號，下面改進一下

def deco(fun=None, param=1):def _deco(fun):def __deco(*args, **kwargs):print (param)fun(*args, **kwargs)return __decoif fun is None:return _decoelif callable(fun):return _deco(fun)else:raise ValueError("Invalid arguments provided to deco")@deco
def f(strx):print (strx)
f('hello')

這樣做，少寫了括號也不怕。

因為你寫了括號，那么此時fun是None，他會去走if分支

如果你忘了寫括號，那么此時fun的值就是函數f這個對象了，不為None他就走elif分支

這就是更兼容的帶參裝飾器。很多三方庫里面的帶參裝飾器都是下面這種寫法。

?

?

?