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

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

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 

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

云服务器提供商
IT科技类资讯
上一篇:质行万里,同舟奋楫 | 新华三 “服务合作伙伴运营能力跃升计划”启动
下一篇:数据中心审计需求的变化