来源:早起Python
作者:刘早起
大家好,我是早起。
之前的文章中提到,很多人认为理解了装饰器的概念和用法后,会觉得自己的 Python 水平有一个明显的提高。
但很多教程在一上来就会给出装饰器的定义以及基本用法,例如你一定会在很多文章中看到例如代码运行时间计时器等相关常用装饰器。
直接从应用入手这样学习当然十分有效,但不是看过就忘就是似懂非懂的状态,因为装饰器从来就不是一个单独的概念,就像数学分析中求积分一样,你可以通过公式快速算出需要求的积分,但是若明白积分是由极限定义的,之后再看积分将会是不一样的视角。
本文我将尝试说清楚为什么需要现装饰器、什么是装饰器、以及如何写一个简单的装饰器,但要彻底理解装饰器还要从函数开始说起,下面是有关函数的四个重要的概念,希望大家可以明白。
相信你在大多数文章中,至少也能知道例如「装饰器是装饰函数」,「在不修改函数代码的情况下增加额外功能」等核心概念,但首先要知道为什么函数能够被装饰。
例如在《流畅的Python》一书中,讲到函数的一开始就提出了一个概念,函数是一等对象
正如书中所说,在Python中一个函数既可以作为参数被传递,也能作为另一个函数的返回值,这也是函数可以被装饰的关键,在介绍装饰器之前,下面有必要通过简单的代码对这段话做一个更直观的理解。
1.1 函数中传递函数
函数中传递函数意思就是可以将函数当作变量来使用,我们来看一个简单的示例。
在下面的代码中,func1是一个普通的函数,接受两个参数a,b并返回他们的和。func2不一样的地方在于多接收了一个func参数,这个func变量需要是一个函数
def func1(a,b): print(f"函数 {func1.__name__} 正在执行") return a + b def func2(func,c,d): print(f"函数 {func2.__name__} 正在执行") return func(c,d)
现在让我们来执行func1
>>> func1(1,2)
函数 func1 正在执行 3
下面func1作为参数执行func2
>>> func2(func1,3,4)
函数 func2 正在执行
函数 func1 正在执行 7
可以看到,先执行func2,在func2接收到fun1后,再次执行func1并返回。注意这里的func1没有括号,它只不过是和a,b一样的参数被使用,理解这点后我们继续看下一个知识点。
1.2 函数中定义函数
在定义一个函数后,可以继续在函数内部定义新的函数。为了理解这点,我们来看下面简单的示例。
我们先定义了一个函数func1,并在func1中定义了func2,并在func1的内部调用了func2
def func1(): print(f"函数 {func1.__name__} 正在执行") def func2(): print(f"内部函数 {func2.__name__} 正在执行")
func2()
现在执行func1与func2看看会发生什么
>>> func1()
函数 func1 正在执行
内部函数 func2 正在执行 >>> func2()
------------------------------------------------
Traceback (most recent call last)
----> 1 func2() NameError: name 'func2' is not defined
可以看到,当执行func1时,会自动执行func2,但是如果单独执行func2,则提示未定义,说明func2只能在func1中被调用!
1.3 函数返回函数
最后是一个函数可以将另一个函数作为返回值返回的简单示例,在下面的代码中,我们先定义了一个外部函数func1(接受一个参数a),之后定义了一个内部函数func2(接受一个参数b)并返回a + b,最后将func2作为func1的返回值返回
def func1(a): print(f"函数 {func1.__name__} 正在执行") def func2(b): print(f"函数 {func2.__name__} 正在执行") return a + b return func2
需要注意的是,这里返回的func2没有括号,代表返回的是func2的地址!
>>> func3 = func1(1) >>> func3
函数 func1 正在执行
<function __main__.func1.<locals>.func2(b)> >>> func3(2)
函数 func2 正在执行 3
从上面的运行结果可以看到,当执行func1(1)后,返回的是func2的地址,并赋给func3,之后执行func3(2)才真正执行了内部函数func2!
现在我们就解决了上一小节的问题「将内部函数func2单独拿出来用」!
1.4 函数内省
函数内省是相对来说比较好理解的一个概念,在Python中的意思就是我们可以访问函数的部分属性,例如print函数,可以使用dir函数来查看其全部属性
>>> dir(print)
['__call__', '__class__', '__delattr__',
··· ··· '__subclasshook__', '__text_signature__']
现在可以查看其对应的属性
>>> print.__name__ 'print' >>> print.__call__
<method-wrapper '__call__' of builtin_function_or_method object at 0x7fddb8056b80> >>> print.__doc__ "print(value, ..., sep=' ', end='\n', file=sys.stdout, flush=False)nnPrints the values to a stream, or to sys.stdout by default.nOptional keyword arguments:nfile: a file-like object (stream); defaults to the current sys.stdout.nsep: string inserted between values, default a space.nend: string appended after the last value, default a newline.nflush: whether to forcibly flush the stream."
函数内省了解到这个程度即可,我们会在2.3节再次提到这里的知识!
至此,我已经将接触装饰器之前必须要吃透的知识点介绍完毕,如果你觉得我讲解的不够清晰,可以查看任何其他教程或者书籍弄懂后再继续阅读。
现在终于可以来说说装饰器了,当然绝对不是直接告诉你一个写好的装饰器,而是我们一点一点去写一个简单的装饰器。
2.1 第一个装饰器
在下面的代码中,我们先定义了一个函数first_decorator,该函数接受函数为参数(如果不理解请查看本文 1.1 节),之后在内部定义了一个名为name_wrapper的内部函数(如果不理解请查看本文 1.2 节),最后返回以name_wrapper作为返回值(如果不理解请查看本文 1.3 节)
def first_decorator(func): def name_wrapper(): print(f"被装饰的函数 {func.__name__} 即将执行")
func()
print(f"被装饰的函数 {func.__name__} 执行完毕") return name_wrapper
这个函数的功能是,在执行被接收函数前后分别打印一段话,所以我们要再定义一个函数来测试效果
def add(): print("函数 add 正在执行 ")
这个fun1没有什么好说的,打印一段话。下面需要仔细看了,我们来执行这两段代码
>>> add = first_decorator(add)
>>> add()
被装饰的函数 add 即将执行
函数 add 正在执行
被装饰的函数 add 执行完毕
正如我们预料的一样,在执行add前后都有一段提示,但是如果每次使用first_decorator功能都需要先将add传递,之后再调用,来回写好几遍,实在太麻烦了!
因此这完全有更Pythonic的写法,也就是我们常见的装饰器形式,使用语法糖@,例如上面的例子和下面的写法等价
@first_decorator def add(): print("函数 add 正在执行 ")
用@+装饰器函数名字放在需要被装饰函数的上方即可,现在直接调用add即可实现装饰器的功能!
>>> add()
被装饰的函数 add 即将执行
函数 add 正在执行
被装饰的函数 add 执行完毕
相信看到这里,你应该明白装饰器@如何工作的,至少你在未来看到@时需要想到类似等价于add = first_decorator(add)一样的作用!
2.2 装饰器传参
上面仅是最简单的装饰器示例,在实际使用时
很自然的想法那就是加上参数呗,改起来也不难
@first_decorator def add(x,y): print("函数 add 正在执行 ")
print(f"{x} + {y} 的结果为{x+y}")
让我们来测试一下
>>> add(1,2)
-----------------------------------------------
Traceback (most recent call last)
<ipython-input-144-944f4051a32c> in <module> ----> 1 add(1,2) TypeError: name_wrapper() takes 0 positional arguments but 2 were given
不出意外的报错了,虽然我们给被装饰的函数加上了参数,但是在装饰器的内部函数name_wrapper()执行时并没有参数!
因此我们之前的代码可以这么改,使用*args, **kwargs也是非常常见的用法
def first_decorator(func): def name_wrapper(*args, **kwargs): print(f"被装饰的函数 {func.__name__} 即将执行")
func(*args, **kwargs)
print(f"被装饰的函数 {func.__name__} 执行完毕") return name_wrapper
现在我们再次使用这个装饰器即可返回我们预测的结果!
@first_decorator def add(x,y): print("函数 add 正在执行 ")
print(f"{x} + {y} 的结果为 {x+y}") >>> add(1,2)
被装饰的函数 add 即将执行
函数 add 正在执行 1 + 2 的结果为 3 被装饰的函数 add 执行完毕
本文的最后,还需要简单介绍一下在写装饰器时常用的functools模块。
还记得1.4节的函数内省相关知识吗?我们可以打印一个函数指向的内存地址或者名字等其他属性。
还是上面用到的add函数,我们都知道虽然被装饰了,但是功能上没有任何变化,依旧是计算两个数字的和,但是真的没有任何变化吗?下面让我们来观察一下
def add(x,y): print("函数 add 正在执行 ")
print(f"{x} + {y} 的结果为{x+y}") >>> print(add)
<function add at 0x7fddb9dd41f0>
>>>print(add.__name__)
add @first_decorator def add(x,y): print("函数 add 正在执行 ")
print(f"{x} + {y} 的结果为{x+y}") >>> print(add)
<function first_decorator.<locals>.name_wrapper at 0x7fddb9dd4e50> >>> print(add.__name__)
name_wrapper
可以看到,被装饰后,虽然功能上没有变化,但是它指向的是装饰器所定义的内部函数!这并不是我们希望看到的,比如若不同函数被两个装饰器装饰时则会出现一样的函数名!
幸运的是Python中的functools库可以轻松解决这个问题,只需要加上一行简单的代码就可以搞定!
import functools def first_decorator(func): @functools.wraps(func) def name_wrapper(*args, **kwargs): print(f"被装饰的函数 {func.__name__} 即将执行")
func(*args, **kwargs)
print(f"被装饰的函数 {func.__name__} 执行完毕") return name_wrapper @first_decorator def add(x,y): print("函数 add 正在执行 ")
print(f"{x} + {y} 的结果为{x+y}") >>> print(add)
<function add at 0x7fddb9dd4e50> >>> print(add.__name__)
add
可以看到此时函数名等函数属性均保留下来了,事实上@functools.wraps(func)通过functools.update_wrapper()将原函数中的部分内省属性固定,只传递部分关键参数来实现这个功能,感兴趣的读者可以自行进一步研究。
至此,我想你应该明白为什么需要现装饰器、什么是装饰器、以及如何写一个简单的装饰器,当你再次看到装饰器时,脑海中浮现的概念应该不仅仅是@。有关装饰器更高级的用法,以及一些常见、好用的装饰器,我将在装饰器的第二篇文章中进行介绍!
数据分析咨询请扫描二维码
若不方便扫码,搜微信号:CDAshujufenxi
你是否被统计学复杂的理论和晦涩的公式劝退过?别担心,“山有木兮:统计学极简入门(Python)” 将为你一一化解这些难题。课程 ...
2025-03-31在电商、零售、甚至内容付费业务中,你真的了解你的客户吗? 有些客户下了一两次单就消失了,有些人每个月都回购,有些人曾经是 ...
2025-03-31在数字化浪潮中,数据驱动决策已成为企业发展的核心竞争力,数据分析人才的需求持续飙升。世界经济论坛发布的《未来就业报告》, ...
2025-03-28你有没有遇到过这样的情况?流量进来了,转化率却不高,辛辛苦苦拉来的用户,最后大部分都悄无声息地离开了,这时候漏斗分析就非 ...
2025-03-27TensorFlow Datasets(TFDS)是一个用于下载、管理和预处理机器学习数据集的库。它提供了易于使用的API,允许用户从现有集合中 ...
2025-03-26"不谋全局者,不足谋一域。"在数据驱动的商业时代,战略级数据分析能力已成为职场核心竞争力。《CDA二级教材:商业策略数据分析 ...
2025-03-26当你在某宝刷到【猜你喜欢】时,当抖音精准推来你的梦中情猫时,当美团外卖弹窗刚好是你想吃的火锅店…… 恭喜你,你正在被用户 ...
2025-03-26当面试官问起随机森林时,他到底在考察什么? ""请解释随机森林的原理""——这是数据分析岗位面试中的经典问题。但你可能不知道 ...
2025-03-25在数字化浪潮席卷的当下,数据俨然成为企业的命脉,贯穿于业务运作的各个环节。从线上到线下,从平台的交易数据,到门店的运营 ...
2025-03-25在互联网和移动应用领域,DAU(日活跃用户数)是一个耳熟能详的指标。无论是产品经理、运营,还是数据分析师,DAU都是衡量产品 ...
2025-03-24ABtest做的好,产品优化效果差不了!可见ABtest在评估优化策略的效果方面地位还是很高的,那么如何在业务中应用ABtest? 结合企业 ...
2025-03-21在企业数据分析中,指标体系是至关重要的工具。不仅帮助企业统一数据标准、提升数据质量,还能为业务决策提供有力支持。本文将围 ...
2025-03-20解锁数据分析师高薪密码,CDA 脱产就业班助你逆袭! 在数字化浪潮中,数据驱动决策已成为企业发展的核心竞争力,数据分析人才的 ...
2025-03-19在 MySQL 数据库中,查询一张表但是不包含某个字段可以通过以下两种方法实现:使用 SELECT 子句以明确指定想要的字段,或者使 ...
2025-03-17在当今数字化时代,数据成为企业发展的关键驱动力,而用户画像作为数据分析的重要成果,改变了企业理解用户、开展业务的方式。无 ...
2025-03-172025年是智能体(AI Agent)的元年,大模型和智能体的发展比较迅猛。感觉年初的deepseek刚火没多久,这几天Manus又成为媒体头条 ...
2025-03-14以下的文章内容来源于柯家媛老师的专栏,如果您想阅读专栏《小白必备的数据思维课》,点击下方链接 https://edu.cda.cn/goods/sh ...
2025-03-13以下的文章内容来源于刘静老师的专栏,如果您想阅读专栏《10大业务分析模型突破业务瓶颈》,点击下方链接 https://edu.cda.cn/go ...
2025-03-12以下的文章内容来源于柯家媛老师的专栏,如果您想阅读专栏《小白必备的数据思维课》,点击下方链接 https://edu.cda.cn/goods/sh ...
2025-03-11随着数字化转型的加速,企业积累了海量数据,如何从这些数据中挖掘有价值的信息,成为企业提升竞争力的关键。CDA认证考试体系应 ...
2025-03-10