文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2023-05-14 20:57

关注

一、闭包

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

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

def make_average():
# 创建一个列表,用来保存数值
nums = []

# 定义一个内部函数,用来计算列表的平均值
def average(n):
# 将数值添加到列表中
nums.append(n)
# 返回平均值
return sum(nums) / len(nums)

return average
  1. 首先,定义一个函数make_average;
  2. 其次,在make_average函数内定义一个空列表,用来存储数值;
  3. 再次,定义一个内部函数,用来计算列表平均值;
  4. 最后,将这个内函数作为外函数make_average的返回值,注意不要加( ),加( )就变成调用这个函数了。
# 调用外部函数,并将其复制给一个变量,注意:此时返回的是内函数的内存地址
a = make_average()
# 给这个变量加(),就相当于调用了内函数average
print(a(20))
print(a(30))

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

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

二、装饰器

1.装饰器引入

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

def add(a, b):
"""计算两数之和"""
res = a + b
return res

def mul(a, b):
"""计算两数之积"""
res = a * b
return res

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

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

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

def new_add(a, b):
print("开始计算...")
r = add(a, b)
print("计算结束...")
return r


print(new_add(22, 33))

执行结果如下:

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

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

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

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

2.函数装饰器

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

1)被装饰函数不带参数

例如:

def wrapper_info(func):
def inner():
print("开始介绍...")
res = func()
print("介绍结束...")
return res

return inner

def introduce1():
print("我是周润发,我来自HONG KONG")

info = wrapper_info(introduce1)
info()

运行结果如下:

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

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

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

2)被装饰函数带参数

尽管可以在装饰器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}")

运行结果如下:

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

3)装饰器带参数

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

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')

运行结果如下:

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

3.装饰器调用

方式一:以函数方式调用

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装饰器-闭包与函数装饰器的详细内容,更多请关注编程网其它相关文章!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯