Python装饰器1-闭包与函数装饰器

简介: 闭包与函数装饰器:被装饰函数不带参数、被装饰函数带参数、装饰器带参数,装饰器的调用

一、闭包

在学习装饰器前,需要先了解闭包的概念。

形成闭包的要点:

  • 函数嵌套
  • 将内部函数作为外部函数的返回值
  • 内部函数必须要使用到外部函数的变量

下面以一个计算列表平均值的案例来讲解闭包:

defmake_average():
# 创建一个列表,用来保存数值nums= []
# 定义一个内部函数,用来计算列表的平均值defaverage(n):
# 将数值添加到列表中nums.append(n)
# 返回平均值returnsum(nums) /len(nums)
returnaverage
  1. 首先,定义一个函数make_average;
  2. 其次,在make_average函数内定义一个空列表,用来存储数值;
  3. 再次,定义一个内部函数,用来计算列表平均值;
  4. 最后,将这个内函数作为外函数make_average的返回值,注意不要加( ),加( )就变成调用这个函数了。
# 调用外部函数,并将其复制给一个变量,注意:此时返回的是内函数的内存地址a=make_average()
# 给这个变量加(),就相当于调用了内函数averageprint(a(20))
print(a(30))

运行结果如下:当传入的数值为20时,列表中只有一个数,所以计算结果是20;当再传入一个数值30时,此时列表中就有两个数20和30,所以平均值的计算结果是25.

二、装饰器

1.装饰器引入

例如,有以下两个函数,分别计算两个数的和以及成绩:

defadd(a, b):
"""计算两数之和"""res=a+breturnresdefmul(a, b):
"""计算两数之积"""res=a*breturnres

现在有个需求:我想要在每个函数的计算开始前打印“开始计算...”,在计算结束后打印“计算结束...”。我们可以通过直接修改函数代码的方式来满足这个需求,但这样会面临以下问题:

  1. 如果要修改的函数过多,十个甚至一百个函数,未免不现实;
  2. 不便于后期维护,例如我不想打印“开始计算...”了,而是要打印“begin...”,岂不是又要重新修改一遍;
  3. 违反开闭原则(OCP),即程序的设计,要求开放对程序的扩展、关闭对程序的修改;

所以,上述直接修改函数代码的方式不可行。我们希望在不修改原函数的情况下,实现对函数的扩展。例如:

defnew_add(a, b):
print("开始计算...")
r=add(a, b)
print("计算结束...")
returnrprint(new_add(22, 33))

执行结果如下:

这种新创建一个函数的方式虽然没有修改原函数,但面临一个很严重的问题:

只能扩展指定函数,不能通用于其它函数,例如扩展上述的add函数,而不能扩展mul函数,如果想要扩展mul函数,只能再创建一个扩展函数;

因为,我们希望可以定义一个通用的扩展函数,可以作用域所有函数。这类不改变原函数代码的通用函数就是:装饰器

2.函数装饰器

装饰器本质上是一个python函数或类,它可以让其他函数或类在不需要做任何代码修改的前提下增加额外功能,也就是为已经存在的对象添加额外功能,装饰器的返回值也是一个函数/类对象。它经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。

1)被装饰函数不带参数

例如:

defwrapper_info(func):
definner():
print("开始介绍...")
res=func()
print("介绍结束...")
returnresreturninnerdefintroduce1():
print("我是周润发,我来自HONG KONG")
info=wrapper_info(introduce1)
info()

运行结果如下:

可见,在没有改变原函数代码的情况下,即给原函数增加了一些额外的功能,func是要修饰的函数,作为一个变量传入装饰函数,能够通用于其他函数,这个wrapper_info就是装饰器。但目前面临的问题是,被装饰函数如果带参数怎么办?例如:

defintroduce2(name, age):
print(f"我叫{name}, 我今年{age}岁了")

2)被装饰函数带参数

尽管可以在装饰器wrapper_info中传入name、age,但并不是每个被装饰的函数都只有name、age,亦或是指定类型的参数,还有可能传入的是字典、列表、元组等。也就是传入参数的类型和数量不固定怎么办?

此时就需要用到不定长参数:(*args, **kwargs)

defwrapper_info(func):
"""    用来对其他函数进行扩展,使其他函数可以在执行前做一些额外的动作    :param func: 要扩展的函数对象    :return:    """definner(*args, **kwargs):
print("开始介绍...")
res=func(*args, **kwargs)
print("介绍结束...")
returnresreturninner

例如:

defintroduce3(name, age, city):
print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")

运行结果如下:

3)装饰器带参数

上述提到的是装饰器,一种是应用于被装饰的函数不带参数,一种是被装饰的函数带参数,那装饰器本身能否带参数呢?比如我定义一个变量,想通过传入不同的值来控制这个装饰器实现不同的功能。答案是肯定的,例如:

defuse_log(level):
defdecorator(func):
definner(*args, **kwargs):
iflevel=="warn":
logging.warning("%s is running by warning"%func.__name__)
eliflevel=="info":
logging.warning("%s is running by info"%func.__name__)
else:
logging.warning("%s is running by other"%func.__name__)
returnfunc(*args, **kwargs)
returninnerreturndecoratordefintroduce4(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')

运行结果如下:

3.装饰器调用

方式一:以函数方式调用

info3=wrapper_info(introduce3)
info3('刘德华', 28, '香港')

如果是装饰器函数带参数,则调用方式为:

info4=use_log(introduce4('周星驰', 28, '香港'))
info4('info')

方式二:以语法糖方式调用

即在被装饰函数上方以@符号进行修饰

@wrapper_infodefintroduce3(name, age, city):
print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")
introduce3('刘德华', 28, '香港')

如果是装饰器函数带参数,例如上述的use_log,则需要在装饰器中传入参数:

@use_log('info')
defintroduce4(name, age, city):
print(f"我叫{name}, 我今年{age}岁了, 我来自{city}")

小结

什么是装饰器?

在不改变原函数代码的情况下,给原函数增加了一些额外的功能,并且能够通用于其他函数,这样的函数就称作为装饰器

装饰器的调用

可以通过传统调用函数的方式进行调用,也可以通过@装饰器的方式调用

装饰器的特点

  • 通过装饰器,可以在不修改原来函数的情况下对函数进行扩展
  • 一个函数可以同时指定多个装饰器




相关文章
|
2天前
|
数据挖掘 数据处理 索引
python常用pandas函数nlargest / nsmallest及其手动实现
python常用pandas函数nlargest / nsmallest及其手动实现
18 0
|
15天前
|
Python
python函数的参数学习
学习Python函数参数涉及五个方面:1) 位置参数按顺序传递,如`func(1, 2, 3)`;2) 关键字参数通过名称传值,如`func(a=1, b=2, c=3)`;3) 默认参数设定默认值,如`func(a, b, c=0)`;4) 可变参数用*和**接收任意数量的位置和关键字参数,如`func(1, 2, 3, a=4, b=5, c=6)`;5) 参数组合结合不同类型的参数,如`func(1, 2, 3, a=4, b=5, c=6)`。
14 1
|
16天前
|
监控 Python
Python中的装饰器:提升代码灵活性与可读性
在Python编程中,装饰器是一种强大的工具,能够提升代码的灵活性和可读性。本文将介绍装饰器的基本概念、使用方法以及实际应用场景,帮助读者更好地理解和利用这一功能。
|
18天前
|
缓存 监控 Python
解密Python中的装饰器:优雅而强大的编程利器
Python中的装饰器是一种强大而又优雅的编程工具,它能够在不改变原有代码结构的情况下,为函数或类添加新的功能和行为。本文将深入解析Python装饰器的原理、用法和实际应用,帮助读者更好地理解和利用这一技术,提升代码的可维护性和可扩展性。
|
2天前
|
存储 缓存 Python
Python装饰器
Python装饰器
12 0
|
3天前
|
Python
深入理解Python中的装饰器
在Python编程中,装饰器(decorators)是一种强大的工具,用于增强函数或类的功能而不改变其原始定义。本文将深入探讨装饰器的概念、用法和实际应用,帮助读者更好地理解和利用这一特性。
|
3天前
|
数据安全/隐私保护 Python
Python中的装饰器:提升代码可读性和灵活性
Python中的装饰器是一种强大的编程工具,能够提升代码的可读性和灵活性。本文将深入探讨装饰器的原理和用法,以及如何利用装饰器来简化代码、实现日志记录、权限控制等功能,从而让你的Python代码更加优雅和高效。
|
7天前
|
Serverless 开发者 Python
《Python 简易速速上手小册》第3章:Python 的函数和模块(2024 最新版)
《Python 简易速速上手小册》第3章:Python 的函数和模块(2024 最新版)
40 1
|
8天前
|
索引 Python
Python高维变量选择:SCAD平滑剪切绝对偏差惩罚、Lasso惩罚函数比较
Python高维变量选择:SCAD平滑剪切绝对偏差惩罚、Lasso惩罚函数比较
10 0
|
8天前
|
数据安全/隐私保护 Python
Python中的装饰器:提升代码可读性与灵活性
Python中的装饰器是一种强大的工具,可以在不改变函数原有逻辑的情况下,为函数添加额外的功能。本文将介绍装饰器的基本概念和用法,并通过实例演示如何利用装饰器提升代码的可读性和灵活性,使代码更加简洁、易于维护。