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)

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

上一篇下一篇

猜你喜欢

热点阅读