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分支

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