第20天:Python 之装饰器

简介: 第20天:Python 之装饰器

1. 概念介绍


装饰器(decorator),又称“装饰函数”,即一种返回值也是函数的函数,可以称之为“函数的函数”。其目的是在不对现有函数进行修改的情况下,实现额外的功能。最基本的理念来自于一种被称为“装饰模式”的设计模式。


在 Python 中,装饰器属于纯粹的“语法糖”,不使用也没关系,但是使用的话能够大大简化代码,使代码更加易读——当然,是对知道这是怎么回事儿的人而言。


想必经过一段时间的学习,大概率已经在 Python 代码中见过@这个符号。没错,这个符号正是使用装饰器的标识,也是正经的 Python 语法。


语法糖:指计算机语言中添加的某种语法,这种语法对语言的功能并没有影响,但是更方便程序员使用。通常来说使用语法糖能够增加程序的可读性,从而减少程序代码出错的机会。


2. 运行机制


简单来说,下面两段代码在语义上是可以划等号的(当然具体过程还是有一点微小区别的):



def IAmDecorator(foo):    '''我是一个装饰函数'''    pass
@IAmDecoratordef tobeDecorated(...):    '''我是被装饰函数'''    pass

与:


def IAmDecorator(foo):    '''我是一个装饰函数'''    pass
def tobeDecorated(...):    '''我是被装饰函数'''    passtobeDecorated = IAmDecorator(tobeDecorated)


可以看到,使用装饰器的@语法,就相当于是将具体定义的函数作为参数传入装饰器函数,而装饰器函数则经过一系列操作,返回一个新的函数,然后再将这个新的函数赋值给原先的函数名。


最终得到的是一个与我们在代码中显式定义的函数同名异质的新函数。


而装饰函数就好像为原来的函数套了一层壳。如图所示,最后得到的组合函数即为应用装饰器产生的新函数:


微信图片_20220209190211.gif


这里要注意一点,上述两段代码在具体执行上还是存在些微的差异。在第二段代码中,函数名tobeDecorated实际上是先指向了原函数,在经过装饰器修饰之后,才指向了新的函数;但第一段代码的执行就没有这个中间过程,直接得到的就是名为tobeDecorated的新函数。


此外,装饰函数有且只能有一个参数,即要被修饰的原函数。


3. 用法


Python 中,装饰器分为两种,分别是“函数装饰器”和“类装饰器”,其中又以“函数装饰器”最为常见,“类装饰器”则用得很少。


3.1 函数装饰器


3.1.1 大体结构


对装饰函数的定义大致可以总结为如图所示的模板,即:


微信图片_20220209190231.jpg


由于要求装饰函数返回值也为一个函数的缘故,为了在原函数的基础上对功能进行扩充,并且使得扩充的功能能够以函数的形式返回,因此需要在装饰函数的定义中再定义一个内部函数,在这个内部函数中进一步操作。最后return的对象就应该是这个内部函数对象,也只有这样才能够正确地返回一个附加了新功能的函数。


如图一的动图所示,装饰函数就像一个“包装”,将原函数装在了装饰函数的内部,从而通过在原函数的基础上附加功能实现了扩展,装饰函数再将这个新的整体返回。同时对于原函数本身又不会有影响。这也是“装饰”二字的含义。


这个地方如果不定义“内部函数”行不行呢?


答案是“不行”。


3.1.2 关于结构的解释


让我们来看看下面这段代码:



>>> def IAmFakeDecorator(fun):...     print("我是一个假的装饰器")...     return fun...>>> @IAmFakeDecorator... def func():...     print("我是原函数")...我是一个假的装饰器

有点奇怪,怎么刚一定义,装饰器扩展的操作就执行了呢?

再来调用一下新函数:


>>> func()我是原函数

诶呦奇了怪了,扩展功能哪儿去了呀?


不要着急,我们来分析一下上面的代码。在装饰函数的定义中,我们没有另外定义一个内部函数,扩展操作直接放在装饰函数的函数体中,返回值就是传入的原函数。

在定义新函数的时候,下面两段代码又是等价的:


>>> @IAmFakeDecorator... def func():...     print("我是原函数")...我是一个假的装饰器


>>> def func():...     print("我是原函数")...>>> func = IAmFakeDecorator(func)我是一个假的装饰器

审视一下后一段代码,我们可以发现,装饰器只在定义新函数的同时调用一次,之后新函数名引用的对象就是装饰器的返回值了,与装饰器没有半毛钱关系。


换句话说,装饰器本身的函数体中的操作都是当且仅当函数定义时,才会执行一次,以后再以新函数名调用函数,执行的只会是内部函数的操作。所以到实际调用新函数的时候,得到的效果跟原函数没有任何区别。


如果不定义内部函数,单纯返回传入的原函数当然也是可以的,也符合装饰器的要求;但却得不到我们预期的结果,对原函数扩展的功能无法复用,只是一次性的。因此这样的行为没有任何意义。


这个在装饰函数内部定义的用于扩展功能的函数可以随意取名,但一般约定俗成命名为wrapper,即“包装”之意。


正确的装饰器定义应如下所示:


>>> def IAmDecorator(fun):...     def wrapper(*args, **kw):...         print("我真的是一个装饰器")...         return fun(*args, **kw)...     return wrapper...


3.1.3 参数设置的问题


内部函数参数设置为(*args, **kw)的目的是可以接收任意参数,关于如何接收任意参数的内容在前面的函数参数[1]部分已经介绍过。


之所以要让wrapper能够接收任意参数,是因为我们在定义装饰器的时候并不知道会用来装饰什么函数,具体函数的参数又是什么情况;定义为“可以接收任意参数”能够极大增强代码的适应性。


另外,还要注意给出参数的位置。


要明确一个概念:除了函数头的位置,其他地方一旦给出了函数参数,表达式的含义就不再是“一个函数对象”,而是“一次函数调用”。


因此,我们的装饰器目的是返回一个函数对象,返回语句的对象一定是不带参数的函数名;在内部函数中,我们是需要对原函数进行调用,因此需要带上函数参数,否则,如果内部函数的返回值还是一个函数对象,就还需要再给一组参数才能够调用原函数。Show code:



>>> def IAmDecorator(fun):...     def wrapper(*args, **kw):...         print("我真的是一个装饰器")...         return fun...     return wrapper...>>> @IAmDecorator... def func(h):...     print("我是原函数")...>>> func()我真的是一个装饰器<function func at 0x000001FF32E66950>

原函数没有被成功调用,只是得到了原函数对应的函数对象。只有进一步给出了下一组参数,才能够发生正确的调用(为了演示参数的影响,在函数func的定义中增加了一个参数h):


>>> func()(h=1)我真的是一个装饰器我是原函数


只要明白了带参数和不带参数的区别,并且知道你想要的到底是什么效果,就不会在参数上犯错误了。并且也完全不必拘泥上述规则,也许你要的就是一个未经调用的函数对象呢?


把握住这一点,嵌套的装饰器、嵌套的内部函数这些也就都不是问题了。


3.1.4 函数属性


本小节内容启发于廖雪峰的官方网站-Python 教程-函数式编程-装饰器[2]

还应注意的是,经过装饰器的修饰,原函数的属性也发生了改变。


>>> def func():...     print("我是原函数")...>>> func.__name__'func'

正常来说,定义一个函数,其函数名称与对应的变量应该是一致的,这样在一些需要以变量名标识、索引函数对象时才能够避免不必要的问题。但是事情并不是那么顺利:


>>> @IAmDecorator... def func():...     print("我是原函数")...>>> func.__name__'wrapper'

变量名还是那个变量名,原函数还是那个原函数,但是函数名称却变成了装饰器中内部函数的名称。


在这里我们可以使用 Python 内置模块functools中的wraps工具,实现“在使用装饰器扩展函数功能的同时,保留原函数属性”这一目的。这里functools.wraps本身也是一个装饰器。运行效果如下:


>>> import functools>>> # 定义保留原函数属性的装饰器... def IAmDecorator(fun):...     @functools.wraps(fun)...     def wrapper(*args, **kw):...         print("我真的是一个装饰器")...         return fun(*args, **kw)...     return wrapper...>>> @IAmDecorator... def func():...     print("我是原函数")...>>> func.__name__'func'

大功告成!


3.2 类装饰器


本节部分参考[Python3 文档-复合语句-类定义[3]]和[python 一篇文章搞懂装饰器所有用法[4]]中类装饰器相关部分


类装饰器的概念与函数装饰器类似,使用上语法也差不多:


@ClassDecoratorclass Foo:    pass

等价于



class Foo:    passFoo = ClassDecorator(Foo)


在定义类装饰器的时候,要保证类中存在__init____call__两种方法。其中__init__方法用以接收原函数或类,__call__方法用以实现装饰逻辑。


简单来讲,__init__方法负责在初始化类实例的时候,将传入的函数或类绑定到这个实例上;而__call__方法则与一般的函数装饰器差不多,连构造都没什么两样,可以认为__call__方法就是一个函数装饰器,因此不再赘述。


3.3 多个装饰器的情况


多个装饰器可以嵌套,具体情况可以理解为从下往上结合的复合函数;或者也可以理解为下一个装饰器的值是前一个装饰器的参数。


举例来说,下面两段代码是等价的:


def func():     passfunc = f1(arg)(f2(func))


理解了前面的内容,这种情况也很容易掌握。


4. 总结


本文介绍了 Python 中的装饰器这一特性,详细讲解了装饰器的实际原理和使用方式,能够大大帮助学习者掌握有关装饰器的知识,减小读懂 Python 代码的阻力,写出更加 pythonic 的代码。


参考

[1] Python3 术语表-装饰器[5]

[2] Python3 文档-复合语句-函数定义[6]

[3] Python3 文档-复合语句-类定义[7]

[4] 语法糖[8]

[5] 廖雪峰的官方网站-Python 教程-函数式编程-装饰器[9]

参考资料

[1]

函数参数: http://www.justdopython.com/2019/09/19/python-function-param-017/

[2]

廖雪峰的官方网站-Python 教程-函数式编程-装饰器: https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584

[3]

[Python3 文档-复合语句-类定义: https://docs.python.org/3/reference/compound_stmts.html#class-definitions

[4]

[python 一篇文章搞懂装饰器所有用法: https://www.jb51.net/article/168276.htm

[5]

1] [Python3 术语表-装饰器: https://docs.python.org/3/glossary.html#term-decorator

[6]

2] [Python3 文档-复合语句-函数定义: https://docs.python.org/3/reference/compound_stmts.html#function-definitions

[7]

3] [Python3 文档-复合语句-类定义: https://docs.python.org/3/reference/compound_stmts.html#class-definitions

[8]

4] [语法糖: https://baike.baidu.com/item/%E8%AF%AD%E6%B3%95%E7%B3%96/5247005?fr=aladdin

[9]

5] [廖雪峰的官方网站-Python 教程-函数式编程-装饰器: https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584

系列文章

目录
相关文章
|
16天前
|
开发者 Python
探索Python中的装饰器:从基础到高级应用
本文将带你深入了解Python中的装饰器,这一强大而灵活的工具。我们将一起探讨装饰器的基本概念,它们如何工作,以及如何使用它们来增强函数和类的功能,同时不改变其核心逻辑。通过具体代码示例,我们将展示装饰器的创建和使用,并探索一些高级应用,比如装饰器堆栈和装饰带参数的装饰器。无论你是初学者还是有经验的开发者,这篇文章都将为你提供新的视角,帮助你更有效地使用装饰器来简化和优化你的代码。
|
17天前
|
测试技术 数据安全/隐私保护 开发者
探索Python中的装饰器:从基础到高级应用
装饰器在Python中是一个强大且令人兴奋的功能,它允许开发者在不修改原有函数代码的前提下增加额外的功能。本文将通过具体代码示例,带领读者从装饰器的基础概念入手,逐步深入到高级用法,如带参数的装饰器和装饰器嵌套等。无论你是初学者还是有经验的开发者,这篇文章都将为你提供有价值的见解和技巧。
|
17天前
|
开发框架 数据建模 中间件
Python中的装饰器:简化代码,增强功能
在Python的世界里,装饰器是那些静悄悄的幕后英雄。它们不张扬,却能默默地为函数或类增添强大的功能。本文将带你了解装饰器的魅力所在,从基础概念到实际应用,我们一步步揭开装饰器的神秘面纱。准备好了吗?让我们开始这段简洁而富有启发性的旅程吧!
25 6
|
5天前
|
缓存 数据安全/隐私保护 Python
python装饰器底层原理
Python装饰器是一个强大的工具,可以在不修改原始函数代码的情况下,动态地增加功能。理解装饰器的底层原理,包括函数是对象、闭包和高阶函数,可以帮助我们更好地使用和编写装饰器。无论是用于日志记录、权限验证还是缓存,装饰器都可以显著提高代码的可维护性和复用性。
20 5
|
19天前
|
测试技术 Python
探索Python中的装饰器:简化代码,增强功能
在Python的世界中,装饰器是那些能够为我们的代码增添魔力的小精灵。它们不仅让代码看起来更加优雅,还能在不改变原有函数定义的情况下,增加额外的功能。本文将通过生动的例子和易于理解的语言,带你领略装饰器的奥秘,从基础概念到实际应用,一起开启Python装饰器的奇妙旅程。
33 11
|
15天前
|
测试技术 开发者 Python
探索Python中的装饰器:从入门到实践
装饰器,在Python中是一块强大的语法糖,它允许我们在不修改原函数代码的情况下增加额外的功能。本文将通过简单易懂的语言和实例,带你一步步了解装饰器的基本概念、使用方法以及如何自定义装饰器。我们还将探讨装饰器在实战中的应用,让你能够在实际编程中灵活运用这一技术。
36 7
|
15天前
|
Python
探索Python中的装饰器:简化代码,增强功能
在Python的世界里,装饰器就像是给函数穿上了一件神奇的外套,让它们拥有了超能力。本文将通过浅显易懂的语言和生动的比喻,带你了解装饰器的基本概念、使用方法以及它们如何让你的代码变得更加简洁高效。让我们一起揭开装饰器的神秘面纱,看看它是如何在不改变函数核心逻辑的情况下,为函数增添新功能的吧!
|
15天前
|
程序员 测试技术 数据安全/隐私保护
深入理解Python装饰器:提升代码重用与可读性
本文旨在为中高级Python开发者提供一份关于装饰器的深度解析。通过探讨装饰器的基本原理、类型以及在实际项目中的应用案例,帮助读者更好地理解并运用这一强大的语言特性。不同于常规摘要,本文将以一个实际的软件开发场景引入,逐步揭示装饰器如何优化代码结构,提高开发效率和代码质量。
42 6
|
15天前
|
存储 缓存 Python
Python中的装饰器深度解析与实践
在Python的世界里,装饰器如同一位神秘的魔法师,它拥有改变函数行为的能力。本文将揭开装饰器的神秘面纱,通过直观的代码示例,引导你理解其工作原理,并掌握如何在实际项目中灵活运用这一强大的工具。从基础到进阶,我们将一起探索装饰器的魅力所在。
|
15天前
|
测试技术 开发者 Python
深入理解Python装饰器:从基础到高级应用
本文旨在为读者提供一个全面的Python装饰器指南,从其基本概念讲起,逐步深入探讨其高级应用。我们将通过实例解析装饰器的工作原理,并展示如何利用它们来增强函数功能、控制程序流程以及实现代码的模块化。无论你是Python初学者还是经验丰富的开发者,本文都将为你提供宝贵的见解和实用的技巧,帮助你更好地掌握这一强大的语言特性。
28 4
下一篇
DataWorks