装饰器
为什么要使用装饰器?目的?
假设我们要增强一个函数A的功能,但又不希望修改函数A的定义,这种在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)。说白了,就是通过新写一个函数B将原函数A包装进去,并添加一些新的代码增加额外的功能。
如何实现?
实现基础原理:由于函数也是一个对象,而且函数对象可以被赋值给变量,所以,通过变量也能调用该函数。
例:设计一个decorator,它可作用于任何函数上,并打印该函数的执行时间:
import functools
def metric(fn):
@functools.wraps(fn)
def logtime(*args, **kw):
print('%s executed in %s ms' % (fn.__name__, 10.24))
return fn(*args, **kw)
return(logtime)
如何调用装饰器?
@metric
def fast(x, y):
time.sleep(0.0012)
return x + y;
只需在函数前一行@装饰器即可,如果需要传入参数则需要多一层函数的装饰器,参考下方链接
注意事项
1、Python内置的functools.wraps可以将fast函数的属性复制到logtime函数中,要在返回函数的函数(即logtime函数)前面加上一句@functools.wraps(fn)
2、fast(x, y)函数的参数x,y通过logtime(args, **kw)中的args, **kw进行传入
参考博客:https://www.liaoxuefeng.com/wiki/1016959663602400/1017451662295584