学科分类
目录
基础

带有参数的装饰器

前面我们讲解的装饰器,都是不带参数的,这些装饰器最终返回的都是函数名。如果我们给装饰器添加参数,那么需要增加一层封装,先传递参数,然后再传递函数名。下面看一段代码:

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)

看到这行代码,大家应该很熟悉了,这就是前面我们学习的无参数的装饰器。相比无参数的装饰器,带参数的装饰器只是用来“加强装饰”的,如果希望装饰器可以根据参数的不同,对不同的函数进行不同的装饰,那么带参数的装饰器是个不错的选择。

点击此处
隐藏目录