爆资讯爆资讯爆资讯

这样用装饰器,为什么不行?

最近几周,陆续收到几位读者关于装饰器使用的提问,今天统一回复。

1 问题

大概问题是这样,想要自定义一个Python装饰器,问我这样写装饰器行不行?如果不行,那又是为什么?

import datetimeimport timedef print_time(g):def f:print('开始执行时间')print(datetime.datetime.today)gprint('结束时间')print(datetime.datetime.today)f

下面使用 print_time装饰函数foo:

@print_timedef foo:time.sleep(2)print('hello world')

当调用 foo函数时,抛出如下异常:

foo---------------------------------------------------------------------------TypeError Traceback (most recent call last) in ----> 1 fooTypeError: '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 datetimeimport timedef print_time(g):def f:print('开始执行时间')print(datetime.datetime.today)gprint('结束时间')print(datetime.datetime.today)return f

装饰 foo:

@print_timedef foo:time.sleep(2)print('hello world')

调用 foo,运行结果如下:

foo开始执行时间2021-04-02 22:32:49.114124hello world结束时间2021-04-02 22:32:51.119506

一切正常

4 装饰器好处

上面自定义print_time装饰器,除了能装饰foo函数外,还能装饰任意其他函数和类内方法。

装饰任意一个函数 foo2

@print_timedef 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

为类MyClassfoo3方法增加print_time装饰:

class MyClass(object):@print_timedef foo3(self):print('this is a method of class')

执行结果如下:

MyClass.foo3开始执行时间2021-04-02 23:16:32.094025this is a method of class结束时间2021-04-02 23:16:32.094078

以上就是装饰器的通俗解释,平时可以多用用,让我们的代码更加精炼、可读。

这是我的第454篇原创,若有帮助,欢迎点赞、分享。

今天给大家推荐一本机器学习、深度学习的人都应该听说过一本经典教材: 《Pattern Recognition and Machine Learning》,中文译名《模式识别与机器学习》,简称 PRML。 出自微软剑桥研究院实验室主任 Christopher Bishop 大神之手。 对,就是豆瓣评分 9.5 的这本书。

这样用装饰器,为什么不行?

资料获取方法

2. 后台回复关键词: PRML

回复「PRML」即可获取资料

未经允许不得转载:爆资讯 » 这样用装饰器,为什么不行?

相关推荐