#带有参数的装饰器 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() 函数和其他一些附加的东西。
装饰器存在的作用就是给函数已经存在的功能再扩展新的功能。