#带有参数的装饰器
import time
def deco(func):
def wrapper(a,b):
startTime = time.time()
func(a,b)
endTime = time.time()
msecs = (endTime - startTime)*1000
print("time is %d ms" %msecs)
return wrapper
@deco
def func(a,b):
print("hello,here is a func for add :")
time.sleep(1)
print("result is %d" %(a+b))
if __name__ == '__main__':
f = func
f(3,4)
不明白为啥在做函数调用的时候f(3,4),要回到
wrapper(a,b)函数的这一句执行:
startTime = time.time()
原因来了:当
@deco这一句被解释器检测到的时候,开始执行装饰器(函数)deco(func),传进去的参数是func()函数在
内存中的地址,返回的也是func()函数在内存中的地址,return wrapper相当于把func()函数在内存中
的地址返回给被装饰过的func()函数,让wrapper指向被装饰器装饰过的func()函数在内存中的地址,这样
的结局是wrapper()函数就相当于是被装饰了的func()函数了。总体而言,装饰器的作用相当于对func
()函数的地址重新赋值,并让内部的wrapper()函数指向这个地址。而wrapper()函数内部的func()
函数等于原来的(被装饰前的)那个func()函数。wrapper()函数也就是新func()函数包含了旧func()
函数和其他一些附加的东西。
装饰器存在的作用就是给函数已经存在的功能再扩展新的功能。