带有参数的装饰器
前面我们讲解的装饰器,都是不带参数的,这些装饰器最终返回的都是函数名。如果我们给装饰器添加参数,那么需要增加一层封装,先传递参数,然后再传递函数名。下面看一段代码:
def func_arg(args):
def func(function_name):
def func_in():
print('--记录日志-args=%s' % args)
function_name()
return func_in
return func
@func_arg('hello')
def test():
print('---test---')
test()
运行代码,结果如下所示:
--记录日志-args=hello
---test---
上述代码中,“@func_arg('hello')”装饰函数test,等价于:
test = func_arg('hello')(test)
由于函数func_arg的返回值是func函数的引用,也就是函数名func,因此,上述代码等价于:
test = func(test)
看到这行代码,大家应该很熟悉了,这就是前面我们学习的无参数的装饰器。相比无参数的装饰器,带参数的装饰器只是用来“加强装饰”的,如果希望装饰器可以根据参数的不同,对不同的函数进行不同的装饰,那么带参数的装饰器是个不错的选择。