您现在的位置是：首页 > 文章详情

Python-装饰器

日期：2018-08-01点击：333收藏

一、什么是装饰器

装饰器就是用来为被修饰对象添加上新功能的工具

注意：装饰器本身可以是任意可调用对象，被装饰器的对象也可以是任意可调用对象

那为什么要用装饰器呢？

这里我们需要知道开放封闭原则：封闭指的是对修改封闭，对扩展开放

装饰器的实现必须遵循两大原则：

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__)