最近几周,陆续收到几位读者关于装饰器使用的提问,今天统一回复。
大概问题是这样,想要自定义一个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装饰器,肯定是不行的。
要想明白为啥不行,首先要知道装饰器这个语法的本质。其实很简单,@print_time装饰foo函数等于:
- foo = print_time(foo)
就是这一行代码,再也没有其他。
因为上面的 print_time 无返回值,所以赋值给 foo 函数后,foo 函数变为 None,所以当调用 foo() 时抛出 'NoneType' object is not callable
这也就不足为奇了。
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
一切正常
上面自定义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
以上就是装饰器的通俗解释,平时可以多用用,让我们的代码更加精炼、可读。
本文转载自微信公众号「后端Q」,作者conan。转载本文请联系后端Q公众号。 概述 ...
基于阿里巴巴的互联网架构、大数据技术,利用混合云架构打造全新的云化电子税 务...
1.百度是个大骗子,我抄了十几年的满分作文却从未得过满分。 2.学神在刷难题,...
3月24日,腾讯发布2020年Q4及全年财报,其中金融科技及企业服务第四季收入385亿...
前言 微服务成了互联网架构的标配模式,对微服务之间的调用的流量治理和管控就尤...
创业与投资的本质,都是追寻一种能够穿越时空,抵达未来的高效方式。 德勤管理咨...
1.某女生寝室门口贴着一个告示男生与饭盒不得入内,问何解?答曰两者都会搞大女...
作者 | 楚奕 来源 | 阿里技术公众号 这篇文章主要从技术视角介绍下跨平台WebCanv...
1.在报名的路上,我看见远处的学校,轰!的一声没了。希望如此。 2.男:我一直...
背景 有时候我会碰到快速搭建测试服务的需求,比如像这样: 搭建一个 HTTP Servi...