今天学到了python的装饰器,感觉这个东西还是稍微有些复杂,所以记录下来,方便以后的查找。虽然标题是python 3.x的装饰器,但是我也没有怎么用过python 2.x,感觉上应该是和python 2.7在用法上差不多。
现在某个视频公司有一段代码,,代码的主要功能就是看电影。
1 def watchfilm(): 2 print('You are watching film now....') 3 4 watchfil()
运行之后输出:
You are watching film now....
现在要求来了,老板提出看电影之前必须要进行登录。
此时小A想到开放封闭原则,原有的代码不能更改,但是需要进行扩展。于是小A有了以下想法:
1.可以定义一个登录函数,将看电影函数的内存地址作为参数传入
2.运行登录函数,将原先看电影函数的内存地址返回
3.使用新的变量存储原先看电影函数的内存地址,将新变量和原先看电影函数的函数名定义为相同的名字
4.运行新变量的时候就就会找到内存地址,运行原先看电影函数的代码
基于以上想法,小A写出了如下代码:
1 def login(func): 2 print('username&password verification....') 3 return func 4 5 def watchfilm(): 6 print('You are watching film now....') 7 8 watchfilm=login(watchfilm) 9 watchfilm()
运行之后输出:
username&password verification....
You are watching film now....
老板看后很满意,然后说:不行!
因为看电影函数是基础平台,而调用是业务部门再进行调用,如果这样修改,就需要业务部门修改大量的代码。
方案被老板否定了,但是活还得接着干啊。小A一筹莫展。
某一天,小A遇见了小B,向小B倾诉了自己的苦恼,小B说道:“使用python的装饰器啊。”
小B解释到,装饰器实现的东西和你做的东西是一样的,不过装饰器使用语法糖@来装饰函数。装饰器执行时,把被装饰函数的内存作为参数传给装饰器,同时执行装饰器函数,最后返回新定义的函数名,和被装饰的函数名相同,只不过内存地址可不一定就是被装饰的函数的内存地址了。再执行被装饰函数时,其实执行的是新的函数。
小A一听,有门,来了精神。经过了一段时间的研究,于是有了一下代码:
1 def login(func): 2 print('username&password verification....') 3 return func 4 5 @login 6 def watchfilm(): 7 print('You are watching film now....') 8 9 watchfilm()
运行之后输出:
username&password verification....
You are watching film now....
能够满足要求,小A顿时兴奋起来。小A马上又测试了几次,没有问题。就在小A想着可以交差的时候,他发现了一个问题。
1 def login(func): 2 print('username&password verification....') 3 return func 4 5 @login 6 def watchfilm(): 7 print('You are watching film now....') 8 9 print('other operation.....') 10 watchfilm()
运行之后输出:
username&password verification....
other operation.....
You are watching film now...
明明没有调用函数,为什么会有输出呢?登录程序应该是在其它操作之后,看电影之前才对啊?小A看了看程序,马上明白了问题所在,因为程序只要启动,就会执行装饰器函数,执行装饰器函数就会执行登录程序,导致了现在这种情况。
这样不行啊,怎么才能解决呢?
忽然小A灵机一动,想到了一个办法,于是有了代码2.0
1 def login(func): 2 def inner(): 3 print('username&password verification....') 4 func() 5 return inner 6 7 @login 8 def watchfilm(): 9 print('You are watching film now....') 10 11 print('other operation.....') 12 watchfilm()
运行之后输出:
other operation..... username&password verification.... You are watching film now....
完美解决了问题,小A长出了一口气。
第二天,小A赶紧找到老板,说出了自己的想法。
老板满意的点点头,对小A说道:你做的很好,但是现在咱们需求变了,目前还想在看电影之前放一段广告而且看完电影之后推荐几部电视节目,你看咱们能不能实现一下。
小A虎躯一震,擦了擦头上的汗,点头应承了下来。但是小A此时也犯了难,如何使用装饰器进行实现呢?小A想到了小B,于是赶紧去请教小B,一顿酒肉计下来,小B跟小A说,你想想装饰器的原理,如果我在有一层def呢?小A一拍大腿,对啊,我怎么没有想到。
小A赶紧进行设计,于是有了代码3.0
1 def advert(): 2 print('advertisement time...') 3 4 def recommend(): 5 print('recommend TV show...') 6 7 def login(adv,rec): 8 def inner(func): 9 def inner2(): 10 print('username&password verification....') 11 adv() 12 func() 13 rec() 14 return inner2 15 return inner 16 17 @login(advert,recommend) 18 def watchfilm(): 19 print('You are watching film now....') 20 21 print('other operation.....') 22 watchfilm()
运行之后输出:
other operation..... username&password verification.... advertisement time... You are watching film now.... recommend TV show...
装饰器带参数之后会首先运行带参数的函数,本例中是login(advert,recommend),函数返回inner函数的内存地址,之后装饰器开始执行,就会变成执行@inner装饰器,之后因为之前两个函数已经传入过,所以再调用函数的时候就能显示输出了。
小A很高兴,这次真是又快又好的完成了任务,于是他唯恐事件有变,赶在老板下班之前向老板进行了汇报。
老板听了小A的汇报相当满意,高声称赞小A,忽然话锋一转,说道:但是小A啊,咱们的需求现在。。。。。。。
(未完待续)