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