这样用装饰器,为什么不行?
2023-03-09 22:11:58 时间
最近几周,陆续收到几位读者关于装饰器使用的提问,今天统一回复。
1. 问题
大概问题是这样,想要自定义一个Python装饰器,问我这样写装饰器行不行?如果不行,那又是为什么?
- import datetime
- import time
- def print_time(g):
- def f():
- print('开始执行时间')
- print(datetime.datetime.today())
- g()
- print('结束时间')
- print(datetime.datetime.today())
- f()
下面使用 print_time装饰函数 foo:
- @print_time
- def foo():
- time.sleep(2)
- print('hello world')
当调用 foo函数时,抛出如下异常:
- foo()
- ---------------------------------------------------------------------------
- TypeError Traceback (most recent call last)
- <ipython-input-27-c19b6d9633cf> in <module>
- ----> 1 foo()
- TypeError: 'NoneType' object is not callable
所以,按照如上定义 print_time装饰器,肯定是不行的。
2. 为什么不行
要想明白为啥不行,首先要知道装饰器这个语法的本质。其实很简单,@print_time装饰foo函数等于:
- foo = print_time(foo)
就是这一行代码,再也没有其他。
因为上面的 print_time 无返回值,所以赋值给 foo 函数后,foo 函数变为 None,所以当调用 foo() 时抛出 'NoneType' object is not callable
这也就不足为奇了。
3. 应该怎么写
print_time 需要返回一个函数,这样赋值给 foo函数后,正确写法如下所示:
- import datetime
- import time
- def print_time(g):
- def f():
- print('开始执行时间')
- print(datetime.datetime.today())
- g()
- print('结束时间')
- print(datetime.datetime.today())
- return f
装饰 foo:
- @print_time
- def foo():
- time.sleep(2)
- print('hello world')
调用 foo ,运行结果如下:
- foo()
- 开始执行时间
- 2021-04-02 22:32:49.114124
- hello world
- 结束时间
- 2021-04-02 22:32:51.119506
一切正常
4. 装饰器好处
上面自定义print_time装饰器,除了能装饰foo函数外,还能装饰任意其他函数和类内方法。
装饰任意一个函数 foo2:
- @print_time
- def foo2():
- print('this is foo2')
装饰类内方法 foo3,需要稍微修改原来的print_time:
- def print_time(g):
- def f(*args, **kargs):
- print('开始执行时间')
- print(datetime.datetime.today())
- g(*args, **kargs)
- print('结束时间')
- print(datetime.datetime.today())
- return f
为类MyClass中foo3方法增加print_time装饰:
- class MyClass(object):
- @print_time
- def foo3(self):
- print('this is a method of class')
执行结果如下:
- MyClass().foo3()
- 开始执行时间
- 2021-04-02 23:16:32.094025
- this is a method of class
- 结束时间
- 2021-04-02 23:16:32.094078
以上就是装饰器的通俗解释,平时可以多用用,让我们的代码更加精炼、可读。
相关文章
- 用Python做疫情数据分析,多维度解析传播率和趋势,未来是乐观的
- Python的import语句笔记
- 这4种统计代码执行耗时,才足够优雅!
- GreenPlum的那些事《五》——浅谈GPDB中的资源队列
- 前后端分离 Vue + Egg.js + Mysql 的 JS全栈实践。动态菜单,RBAC权限模型,WebSocket实现站内信。已部署到线上!!!
- Pychram
- Spring中的定时器都会了?
- Python自学之路—位运算
- Python自学之路—条件、循环语句
- Python自学之路—变量与运算
- 开始着手用Python写一个游戏脚本
- python爬虫爬取QQ号
- HaaS轻应用之Python篇|阿里云产品内容精选(三十七)
- 作为2021年计算机初学者你必须要知道的上云那些事
- 一张图带你搞懂Node事件循环
- 全国41611个景点,用Python告诉你哪些地方更值得一游!
- python 来查 肯德基 address
- python 批量修改文件名
- 炸裂!上手三天,就在开发板上播放BadApple, 还是Python香
- [Python从零到壹] 一.为什么我们要学Python及基础语法详解