在学习装饰器前,需要先了解闭包的概念。形成闭包的要点:
下面以一个计算列表平均值的案例来讲解闭包:
def make_average(): # 创建一个列表,用来保存数值 nums = [] # 定义一个内部函数,用来计算列表的平均值 def average(n): # 将数值添加到列表中 nums.append(n) # 返回平均值 return sum(nums) / len(nums) return average
# 调用外部函数,并将其复制给一个变量,注意:此时返回的是内函数的内存地址 a = make_average() # 给这个变量加(),就相当于调用了内函数average print(a(20)) print(a(30))
运行结果如下:当传入的数值为20时,列表中只有一个数,所以计算结果是20;当再传入一个数值30时,此时列表中就有两个数20和30,所以平均值的计算结果是25.
例如,有以下两个函数,分别计算两个数的和以及成绩:
def add(a, b): """计算两数之和""" res = a + b return res def mul(a, b): """计算两数之积""" res = a * b return res
现在有个需求:我想要在每个函数的计算开始前打印“开始计算...”,在计算结束后打印“计算结束...”。我们可以通过直接修改函数代码的方式来满足这个需求,但这样会面临以下问题:
所以,上述直接修改函数代码的方式不可行。我们希望在不修改原函数的情况下,实现对函数的扩展。例如:
def new_add(a, b): print("开始计算...") r = add(a, b) print("计算结束...") return r print(new_add(22, 33))
执行结果如下:
这种新创建一个函数的方式虽然没有修改原函数,但面临一个很严重的问题:
只能扩展指定函数,不能通用于其它函数,例如扩展上述的add函数,而不能扩展mul函数,如果想要扩展mul函数,只能再创建一个扩展函数;
因为,我们希望可以定义一个通用的扩展函数,可以作用域所有函数。这类不改变原函数代码的通用函数就是:装饰器。
装饰器本质上是一个python函数或类,它可以让其他函数或类在不需要做任何代码修改的前提下增加额外功能,也就是为已经存在的对象添加额外功能,装饰器的返回值也是一个函数/类对象。它经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。
例如:
def wrapper_info(func): def inner(): print("开始介绍...") res = func() print("介绍结束...") return res return inner def introduce1(): print("我是周润发,我来自HONG KONG") info = wrapper_info(introduce1) info()
运行结果如下:
可见,在没有改变原函数代码的情况下,即给原函数增加了一些额外的功能,func是要修饰的函数,作为一个变量传入装饰函数,能够通用于其他函数,这个wrapper_info就是装饰器。但目前面临的问题是,被装饰函数如果带参数怎么办?例如:
def introduce2(name, age): print(f"我叫{name}, 我今年{age}岁了")
尽管可以在装饰器wrapper_info中传入name、age,但并不是每个被装饰的函数都只有name、age,亦或是指定类型的参数,还有可能传入的是字典、列表、元组等。也就是传入参数的类型和数量不固定怎么办?
此时就需要用到不定长参数:(*args, **kwargs)
def wrapper_info(func): """ 用来对其他函数进行扩展,使其他函数可以在执行前做一些额外的动作 :param func: 要扩展的函数对象 :return: """ def inner(*args, **kwargs): print("开始介绍...") res = func(*args, **kwargs) print("介绍结束...") return res return inner
例如:
def introduce3(name, age, city): print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")
运行结果如下:
上述提到的是装饰器,一种是应用于被装饰的函数不带参数,一种是被装饰的函数带参数,那装饰器本身能否带参数呢?比如我定义一个变量,想通过传入不同的值来控制这个装饰器实现不同的功能。答案是肯定的,例如:
def use_log(level): def decorator(func): def inner(*args, **kwargs): if level == "warn": logging.warning("%s is running by warning" % func.__name__) elif level == "info": logging.warning("%s is running by info" % func.__name__) else: logging.warning("%s is running by other" % func.__name__) return func(*args, **kwargs) return inner return decorator def introduce4(name, age, city): print(f"我叫{name}, 我今年{age}岁了, 我来自{city}") info1 = use_log(introduce4('周星驰', 28, '香港')) info1('info') info2 = use_log(introduce4('周润发', 28, '香港')) info2('warn') info3 = use_log(introduce4('成龙', 28, '香港')) info3('xxx')
运行结果如下:
info3 = wrapper_info(introduce3) info3('刘德华', 28, '香港')
如果是装饰器函数带参数,则调用方式为:
info4 = use_log(introduce4('周星驰', 28, '香港')) info4('info')
即在被装饰函数上方以@符号进行修饰
@wrapper_info def introduce3(name, age, city): print(f"我叫{name}, 我今年{age}岁了, 我来自{city}") introduce3('刘德华', 28, '香港')
如果是装饰器函数带参数,例如上述的use_log,则需要在装饰器中传入参数:
@use_log('info') def introduce4(name, age, city): print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")
在不改变原函数代码的情况下,给原函数增加了一些额外的功能,并且能够通用于其他函数,这样的函数就称作为装饰器。
可以通过传统调用函数的方式进行调用,也可以通过@装饰器的方式调用
以上是Python装饰器-闭包与函数装饰器的详细内容。更多信息请关注PHP中文网其他相关文章!