python带参装饰器的改良版

简单点就是这种

def deco2(param=1):
def _deco2(fun):
def __deco2(*args, **kwargs):
print (param)
fun(*args, **kwargs) return __deco2 return _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 __deco if fun is None:
return _deco
elif 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分支

这就是更兼容的带参装饰器。很多三方库里面的带参装饰器都是下面这种写法。

 
上一篇:Python带括号的计算器


下一篇:Python中,关于调用带参函数,一些需要注意的问题(监听时)