一、什么是装饰器
装饰器就是用来为被修饰对象添加上新功能的工具
注意:装饰器本身可以是任意可调用对象,被装饰器的对象也可以是任意可调用对象
那为什么要用装饰器呢?
这里我们需要知道开放封闭原则:封闭指的是对修改封闭,对扩展开放
装饰器的实现必须遵循两大原则:
1.不修改被装饰对象的源代码
2.不修改被装饰对象的调用方式
装饰的目标:就是在满足1和2的前提下为被修饰对象添加上新功能
简单的装饰器:
import time def index(): print('welcome to index page') time.sleep(3) def outter(func): # func=最原始那个index的内存地址 def wrapper(): start=time.time() func() #最原始的那个index的内存地址() stop=time.time() print('run time is %s' %(stop - start)) return wrapper index=outter(index) #index=outter(最原始那个index的内存地址) #index=wrapper的内存地址 index() #wrapper的内存地址()
上面这种装饰器没有传入参数,下面对他进行升级
def home(name): print('welcome %s to home page' %name) time.sleep(2) return 123 def timmer(func): #func=最原始那个home函数的内地址 def wrapper(*args,**kwargs): #args=('egon',) kwargs={} start=time.time() res=func(*args,**kwargs) #最原始那个home函数的内地址('egon') stop=time.time() print('run time is %s' %(stop - start)) return res return wrapper home=timmer(home) #home=outter(最原始那个home函数的内地址) #home=wrapper函数的内地址 res=home('egon') # res=wrapper函数的内地址('egon')
参数是可以传入了,那还有没有可以优化的地方呢,这个时候就用到了python的语法糖,
def outter(func): def wrapper(*args,**kwargs): #在调用函数前加功能 res=func(*args,**kwargs) #调用被装饰的\也就是最原始的那个函数 #在调用函数后加功能 return res return wrapper @outter #index=outter(index) #index=wrapper def index(): print('welcome to index page') time.sleep(3) index()
装饰器的语法糖:在被装饰对象正上方单独一行写@装饰器的名字
运行原理:python解释器一旦运行到@装饰器的名字,就会调用装饰器,然后将被装饰函数的内存地址当做参数传给装饰器,最后将装饰器调用的结果赋值给原函数名
调用多个装饰器:
def outter1(func1): #func1=wrapper2 print('outter1') def wrapper1(*args,**kwargs): print('wrapper1') res1=func1(*args,**kwargs) #res1=wrapper2(*args,**kwargs) return res1 return wrapper1 def outter2(func2): #func2=最原始的那个index的内存地址 print('outter2') def wrapper2(*args,**kwargs): print('wrapper2') res2=func2(*args,**kwargs) return res2 return wrapper2 @outter1 # index=outter1(wrapper2) #index=wrapper1 @outter2 #outter2(最原始的那个index的内存地址) ===> wrapper2 def index(): print('welcome to index page') time.sleep(3) index() #wrapper1() ''' outter2 outter1 wrapper1 wrapper2 '''
解释顺序是自下往上的,执行顺序从上往下,那打印的结果和我们预想的不一样呢,我们应该理解装饰器函数在被装饰器函数定义好后立即执行
模板:
# 有参装饰器的模板 def outter1(x,y,z): def outter2(func): def wrapper(*args,**kwargs): res=func(*args,**kwargs) return res return wrapper return outter2 # 无参装饰器的模板 def outter(func): def wrapper(*args,**kwargs): res=func(*args,**kwargs) return res return wrapper
补充:
调用方式和源代码都没有修改,是不是所有都很完美了呢?不是其实还有一个需要注释的地方就是注释文档
def outter(func): # @wraps(func) def inner(*args,**kwargs): res = func(*args,**kwargs) return res return inner @outter def index(): ''' zfj 1212 1231 ''' # print('1111') index() print(index.__name__)# inner print(index.__doc__)#None
index的函数名和文档都变成inner的了,没有和原来的index一样,这是我们可以在inner函数中加入
inner.__name__=func.__name__
inner.__doc__= func.__doc__
这样index得到的文档就是原来的样子了,还有一个方法是使用wraps!
from functools import wraps def outter(func): @wraps(func) def inner(*args,**kwargs): res = func(*args,**kwargs) return res # inner.__name__=func.__name__ # inner.__doc__= func.__doc__ return inner @outter def index(): ''' zfj 1212 1231 ''' # print('1111') index() print(index.__name__) print(index.__doc__)
很感谢Nisen的装饰器执行顺序迷思,对我的启发很大
https://segmentfault.com/a/1190000007837364
焚膏油以继晷,恒兀兀以穷年。