装饰器的大致思想是,闭包函数引用了外部函数的参数(在装饰器里是个函数),因此利用闭包函数除了调用外部函数这个参数函数以外,包装一些其他的功能,试图扩展这个原始函数
具体的装饰器的理论不做过多的解释,网上参考资料很多,例如下面这篇。点击这篇博客查看
我简单实现了一个计算函数运行时间的装饰器,利用这段代码解读一下装饰器的定义以及使用方法。
timerFunc是外部函数,这个外部函数的参数是一个函数名,即我们要扩展(包装)的函数,例如下面的addMy函数或者multiMy函数。而我们想要对这些基本函数扩展计算运行时间的功能。因此在timerFunc内定义一个闭包函数wrapper,这个函数里调用addMy并将执行结果返回,然后添加上我们想要的计算运行时间的部分。而timerFunc则返回wrapper函数,即扩展后的函数。因此,timerFunc的返回值实则是一个函数句柄。
#计算函数运行时间的装饰器
def timerFunc(f):
def wrapper(*args, **kwargs):
start = time.time()
res = f(*args, **kwargs)
end = time.time()
print('Lapse: ',end-start)
return res
return wrapper
#加法函数
def addMy(a,b):
return a+b
#乘法函数
@timerFunc #语法糖
def multiMy(a,b):
return a*b
具体的使用如下。在使用装饰器的时候,一般都是像上面乘法函数一样,在要扩展的函数定义之前加上语法糖。那么原函数名则变成了wrapper函数的句柄。通过调用multiMy则会增加wrapper函数中附加的功能。
#使用闭包函数的方式调用
>>> a = timerFunc(addMy)
>>> a = timerFunc(addMy)
>>> a(1,2)
Lapse: 4.291534423828125e-06
3
#使用语法糖的方式
>>> multiMy(2,3)
Lapse: 4.0531158447265625e-06
6