【目录】
一、装饰器的介绍
1、什么是装饰器
2、为何要用装饰器
二、装饰器的实现
1、无参装饰器的实现
2、有参装饰器的实现
*储备知识:(请点击下方标题阅读哦~)
1、名称空间和作用域
一、装饰器的介绍
1、为何要用装饰器
开放封闭原则——
软件的设计应该遵循开放封闭原则,即对扩展是开放的,而对修改是封闭的。
对扩展开放,意味着有新的需求或变化时,可以对现有代码进行扩展,以适应新的情况。
对修改封闭,意味着对象一旦设计完成,就可以独立完成其工作,而不要对其进行修改。
软件包含的所有功能的源代码以及调用方式,都应该避免修改,否则一旦改错,则极有可能产生连锁反应,最终导致程序崩溃,
而对于上线后的软件,新需求或者变化又层出不穷,我们必须为程序提供扩展的可能性,这就用到了装饰器。
2、什么是装饰器
’装饰’代指为被装饰对象添加新的功能,
’器’代指器具/工具,装饰器与被装饰的对象均可以是任意可调用对象。
即 装饰器的作用就是在不修改被装饰对象源代码和调用方式的前提下为被装饰对象添加额外的功能。
装饰器经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等应用场景,装饰器是解决这类问题的绝佳设计,
有了装饰器,就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。
提示:可调用对象有函数,方法或者类,此处我们单以本章主题函数为例,来介绍函数装饰器,并且被装饰的对象也是函数。
函数装饰器,指的是定义一个函数,该函数是用来为其他函数添加额外的功能。
二、函数装饰器的实现
函数装饰器分为:无参装饰器和有参装饰器两种,二者的实现原理一样,都是’函数嵌套+闭包+函数对象’的组合使用的产物。
遵循原则:不修改被装饰对象源代码&调用方式
1、无参装饰器的实现
(请紧跟步伐——)
(0)需求——在不修改index函数的源代码以及调用方式的前提下为其添加统计运行时间的功能
def index(x,y): time.sleep(3) print('index %s %s' %(x,y)) index(111,222) # index(y=111,x=222) #其他调用的参数形式 # index(111,y=222)
(1)解决方案一:失败
问题:没有修改被装饰对象的调用方式,但是修改了其源代码
import time def index(x,y): start=time.time() time.sleep(3) print('index %s %s' %(x,y)) stop = time.time() print(stop - start) index(111,222)
(2)解决方案二:失败
问题:没有修改被装饰对象的调用方式,也没有修改了其源代码,并且加上了新功能,但是代码冗余
import time def index(x,y): time.sleep(3) print('index %s %s' %(x,y)) start=time.time() index(111,222) stop=time.time() print(stop - start) start=time.time() index(111,222) stop=time.time() print(stop - start) start=time.time() index(111,222) stop=time.time() print(stop - start)
(3)解决方案三:失败
问题:解决了方案二代码冗余问题,但带来一个新问题即函数的调用方式改变了
import time def index(x,y): time.sleep(3) print('index %s %s' %(x,y)) def wrapper(): start=time.time() index(111,222) #index本身也存在问题,其参数被限定了;index参数一增减,wrapper的也需要增减——如何写活呢? stop=time.time() print(stop - start) wrapper()
(4)优化方案三的大方向:如何在方案三的基础上,不改变函数的调用方式
方案三的优化一:将index的参数写活了
import time def index(x,y,z): time.sleep(3) print('index %s %s %s' %(x,y,z)) def wrapper(*args,**kwargs): start=time.time() index(*args,**kwargs) # index(3333,z=5555,y=44444);wrapper的参数,是传给index使用的。使用该种方式,就可接收任意形式的参数,从而避免修改wrapper的参数 stop=time.time() print(stop - start) # wrapper(3333,4444,5555) # wrapper(3333,z=5555,y=44444)
方案三的优化二:在优化一的基础上把被装饰对象写活了,原来只能装饰index……
import time def index(x,y,z): time.sleep(3) print('index %s %s %s' %(x,y,z)) def home(name): time.sleep(2) print('welcome %s to home page' %name) def outter(func): # func = index的内存地址 def wrapper(*args,**kwargs): start=time.time() func(*args,**kwargs) # index的内存地址() stop=time.time() print(stop - start) return wrapper
#为何要有该行代码?return wrapper—因为wrapper的内存地址为全局的,现在被装进了一个函数里,放在了‘局部’,因此未传值,需返回它的内存地址 (名字,即为其内存地址) index=outter(index) # index=wrapper的内存地址 home=outter(home) # home=wrapper的内存地址 home('egon') # home(name='egon')
方案三的优化三:将wrapper做的跟被装饰对象一模一样,以假乱真
import time def index(x,y,z): time.sleep(3) print('index %s %s %s' %(x,y,z)) def home(name): time.sleep(2) print('welcome %s to home page' %name) def outter(func): def wrapper(*args,**kwargs): start=time.time() res=func(*args,**kwargs) stop=time.time() print(stop - start) return res return wrapper # 偷梁换柱:home这个名字指向的wrapper函数的内存地址 home=outter(home) res=home('egon') # res=wrapper('egon') print('返回值--》',res)
(5)语法糖——让你开心的语法糖,可以含在嘴里很久很久
如何定义并使用装饰器——
== 定义装饰器
== 使用装饰器:在被装饰对象正上方的单独一行写@装饰器名字
注意:名字后面加括号,就会触发函数体的执行。@装饰器名字,即 被装饰对象名=装饰器名字(被装饰对象名)
括号的优先级高于@,即 @装饰器名字(),先执行括号,后为@,但会报错。
糖栗子1:统计运行时间装饰器 timmer
import time #定义装饰器 def timmer(func): def wrapper(*args,**kwargs): start=time.time() res=func(*args,**kwargs) stop=time.time() print(stop - start) return res return wrapper # 使用装饰器:在被装饰对象正上方的单独一行写@装饰器名字 @timmer # index=timmer(index) def index(x,y,z): time.sleep(3) print('index %s %s %s' %(x,y,z)) @timmer # home=timmer(ome) def home(name): time.sleep(2) print('welcome %s to home page' %name) index(x=1,y=2,z=3) home('egon')
(6)总结无参装饰器模板
模板:
def outter(func): def wrapper(*args,**kwargs): # 1、调用原函数 # 2、为其增加新功能 res=func(*args,**kwargs) return res return wrapper
糖栗子2:登录系统认证功能装饰器 auth
def auth(func): def wrapper(*args, **kwargs): # 1、调用原函数 # 2、为其增加新功能 name = input('your name>>: ').strip() pwd = input('your password>>: ').strip() if name == 'egon' and pwd == '123': res = func(*args, **kwargs) return res else: print('账号密码错误') return wrapper @auth def index(): print('from index') index()
(7)吃完糖,思考一下——叠加多个装饰器时,加载顺序与运行顺序是怎样的?
@deco1 # index=deco1(deco2.wrapper的内存地址) @deco2 # deco2.wrapper的内存地址=deco2(deco3.wrapper的内存地址) @deco3 # deco3.wrapper的内存地址=deco3(index) ——加载装饰器:自下而上 def index(): pass
# 叠加多个装饰器
1. 加载顺序(outter函数的调用顺序):自下而上
2. 执行顺序(wrapper函数的执行顺序):自上而下栗子如下:
def outter1(func1): #func1=wrapper2的内存地址 print('加载了outter1') def wrapper1(*args,**kwargs): print('执行了wrapper1') res1=func1(*args,**kwargs) return res1 return wrapper1 def outter2(func2): #func2=wrapper3的内存地址 print('加载了outter2') def wrapper2(*args,**kwargs): print('执行了wrapper2') res2=func2(*args,**kwargs) return res2 return wrapper2 def outter3(func3): # func3=最原始的那个index的内存地址 print('加载了outter3') def wrapper3(*args,**kwargs): print('执行了wrapper3') res3=func3(*args,**kwargs) return res3 return wrapper3 @outter1 # outter1(wrapper2的内存地址)======>index=wrapper1的内存地址 @outter2 # outter2(wrapper3的内存地址)======>wrapper2的内存地址 @outter3 # outter3(最原始的那个index的内存地址)===>wrapper3的内存地址 def index(): print('from index') print('======================================================') index()
(8)补充一下:from functools import wraps
#偷梁换柱,即将原函数名指向的内存地址偷梁换柱成wrapper函数,所以应该将wrapper做的跟原函数一样才行
糖栗子3:
from functools import wraps def outter(func): @wraps(func) def wrapper(*args, **kwargs): """这个是主页功能""" res = func(*args, **kwargs) # res=index(1,2) return res # 手动将原函数的属性赋值给wrapper函数 # 1、函数wrapper.__name__ = 原函数.__name__ # 2、函数wrapper.__doc__ = 原函数.__doc__ # wrapper.__name__ = func.__name__ # wrapper.__doc__ = func.__doc__ return wrapper @outter # index=outter(index) def index(x,y): """这个是主页功能""" print(x,y) print(index.__name__) print(index.__doc__) #help(index)
2、有参装饰器的实现
(1)知识储备
由于语法糖@的限制,outter函数只能有一个参数,并且该参数只用来接收被装饰对象的内存地址
def outter(func): # func = 函数的内存地址 def wrapper(*args,**kwargs): res=func(*args,**kwargs) return res return wrapper # @outter # index=outter(index) # index=>wrapper @outter # outter(index) def index(x,y): print(x,y)
偷梁换柱之后——
index的参数什么样子,wrapper的参数就应该什么样子
index的返回值什么样子,wrapper的返回值就应该什么样子
index的属性什么样子,wrapper的属性就应该什么样子==》from functools import wraps
(2) 登录系统认证功能——山炮玩法一:
def auth(func,db_type): def wrapper(*args, **kwargs): name=input('your name>>>: ').strip() pwd=input('your password>>>: ').strip() if db_type == 'file': print('基于文件的验证') if name == 'egon' and pwd == '123': res = func(*args, **kwargs) return res else: print('user or password error') elif db_type == 'mysql': print('基于mysql的验证') elif db_type == 'ldap': print('基于ldap的验证') else: print('不支持该db_type') return wrapper # @auth # 账号密码的来源是文件 def index(x,y): print('index->>%s:%s' %(x,y)) # @auth # 账号密码的来源是数据库 def home(name): print('home->>%s' %name) # @auth # 账号密码的来源是ldap def transfer(): print('transfer') index=auth(index,'file') home=auth(home,'mysql') transfer=auth(transfer,'ldap') # index(1,2) # home('egon') # transfer()
山炮玩法二:
def auth(db_type): def deco(func): def wrapper(*args, **kwargs): name=input('your name>>>: ').strip() pwd=input('your password>>>: ').strip() if db_type == 'file': print('基于文件的验证') if name == 'egon' and pwd == '123': res = func(*args, **kwargs) return res else: print('user or password error') elif db_type == 'mysql': print('基于mysql的验证') elif db_type == 'ldap': print('基于ldap的验证') else: print('不支持该db_type') return wrapper return deco deco=auth(db_type='file') @deco # 账号密码的来源是文件 def index(x,y): print('index->>%s:%s' %(x,y)) deco=auth(db_type='mysql') @deco # 账号密码的来源是数据库 def home(name): print('home->>%s' %name) deco=auth(db_type='ldap') @deco # 账号密码的来源是ldap def transfer(): print('transfer') index(1,2) home('egon') transfer()
语法糖—版本:@auth(参数)
def auth(db_type): def deco(func): def wrapper(*args, **kwargs): name = input('your name>>>: ').strip() pwd = input('your password>>>: ').strip() if db_type == 'file': print('基于文件的验证') if name == 'egon' and pwd == '123': res = func(*args, **kwargs) # index(1,2) return res else: print('user or password error') elif db_type == 'mysql': print('基于mysql的验证') elif db_type == 'ldap': print('基于ldap的验证') else: print('不支持该db_type') return wrapper return deco @auth(db_type='file') # @deco # index=deco(index) # index=wrapper def index(x, y): print('index->>%s:%s' % (x, y)) @auth(db_type='mysql') # @deco # home=deco(home) # home=wrapper def home(name): print('home->>%s' % name) @auth(db_type='ldap') # 账号密码的来源是ldap def transfer(): print('transfer') index(1, 2) home('egon') transfer()
(3)有参装饰器模板
def 有参装饰器(x,y,z): def outter(func): def wrapper(*args, **kwargs): res = func(*args, **kwargs) return res return wrapper return outter @有参装饰器(1,y=2,z=3) def 被装饰对象(): pass
参考资料: