更新时间:2022-09-30 来源:黑马程序员 浏览量:
前面我们讲解的装饰器,都是不带参数的,这些装饰器最终返回的都是函数名。在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)
看到这行代码,大家应该很熟悉了,这就是前面我们学习的无参数装饰器。相比无参数的装饰器,带参数的装饰器只是用来“加强装饰”的,如果希望装饰器可以根据参数的不同,对不同的函数进行不同的装饰,那么带参数的装饰器是个不错的选择。