首页 > 代码库 > 浅显易懂的谈一谈python中的装饰器！！
浅显易懂的谈一谈python中的装饰器！！
hello大家好~~我是稀里糊涂林老冷，一天天稀里糊涂的。
前一段时间学习了装饰器，觉着这东西好高大上哇靠！！哈哈，一定要总结一下，方便以后自己查阅，也希望帮助其他伙伴们共同进步！
装饰器：
大家可以这样理解，装饰器是运用闭包的基本原理，对一个目标函数进行装饰。即是在执行一个目标函数之前、之后执行一些特定的事情。
学习装饰器一定要有闭包的基础知识，如果对闭包不是特别理解的话，可以参考我之前的博文http://www.cnblogs.com/Lin-Yi/p/7305364.html，也可以学习其他大神的博文哦~~~一定要弄明白闭包，要不然装饰器很难理解的。
不理解的话举个例子：
工作中，他写了登录功能，我写了注销功能，他写了保存个人信息功能.........你和你的伙伴是一个团队，你们完成各种各样的功能。但是再软件上线的时候呢，技术总监说了，我们要统计每一个功能模块执行耗费的时间，针对耗费时间太多的功能，可能我们要开会商讨进行优化。
问题来了，我们团队五六个人，每个人写了两三个功能，都要统计执行实现，是不是每个人都要修改自己的代码，在功能模块函数开始的时候取一下时间，结束的时候取一下时间，然后做一个减法。每一个人的每一个功能都去增加这样一些代码，是不是可麻烦了呢！是不是可浪费时间了呢！！那怎么办？？ 用装饰器！
我们只需要编写一个装饰器，用这个装饰器对我们写的所有功能进行装饰，就能达到同一个目的！！哈哈哈，很牛有木有！我们一起看看吧~
先上一段代码！！！
1 import time
2 #装饰器会用到闭包的原理：外函数内部定义了一个内函数，内函数使用外函数的局部变量，并且外函数返回了内函数的引用
3 #装饰器函数 传入一个我们想对他装饰的目标函数的引用，将在内函数中使用
4 def decorator( func ):
5 def inner():
6 t1 = time.time() # 目标函数开始之前取一下时间
7 print("登录前的时间是",t1)
8 # 这个func外部函数传入的参数，也就是我们希望装饰的目标函数的引用
9 # 这里实际上执行了目标函数，我们想对这个函数进行装饰，所有在它执行之前和之后进行一番操作，具体什么操作看业务逻辑
10 func()
11 t2 = time.time() # 目标函数结束之后取一下时间
12 print("登录后的时间是",t2)
13 t = t2 - t1 #计算目标函数执行花了多长时间
14 print("登录花费的时间:",t)
15 #在外函数结束时候返回内函数的引用
16 return inner
17
18 # 上面是我们写的装饰器函数，用装饰器函数可以对一个目标函数进行操作！！
19 ‘‘‘
20 下面的 @decorator 就是是使用装饰函数decorator的意思
21 它实际上执行的是: login = decorator( login )
22 注意，编程语言都是从右向左来解析执行，这句话实际上会发生的事情是：
23 1 把目标函数login(是一个变量名，里面存的是目标函数的引用) 传入decorator函数被参数func接收了
24 这时候func也是目标函数的引用 func和login指向同一个函数对象
25 2 decorator函数定义了内部函数inner，发现里面会使用到func，
26 这用到闭包的原理，外函数结束的时候会把func绑定给内函数将来使用
27 3 外函数结束的时候把自己创建的内函数的引用inner返回给了login接收，
28 这时候login已经不是原来我们编写的目标函数了，login实际上是一个inner函数的实例对象
29 再执行login() 的时候实际上执行了inner()的一个对象
30 4 再执行login() 的时候 实际上执行inner()
31 会先执行取时间 打印
32 之后执行func()才是执行我们的目标函数
33 最后又取时间 打印结果
34 ‘‘‘
35 @decorator
36 #下面我们写一个函数模拟是登录功能模块的函数，并且用装饰器进行装饰
37 def login():
38 print("用户正在登录！请稍等。。。。")
39 time.sleep(1)
40 print("登录成功！欢迎登录！")
41
42
43
44 if __name__ == ‘__main__‘:
45 #调用login 实际上调用了装饰之后的函数inner的一个对象，inner里的func才是真正的目标函数
46 login()
47 ‘‘‘
48 执行结果:
49 登录前的时间是 1502163280.2836263
50 用户正在登录！请稍等。。。。
51 登录成功！欢迎登录！
52 登录后的时间是 1502163281.2895217
53 登录花费的时间: 1.0058953762054443
54 ‘‘‘
不知道大家有没有看懂这段代码。认真读一读注释的话，我想有闭包基础的伙伴们一定能读懂装饰器。
总之装饰器很神奇，运用引用之间互相改变，实现了一段很厉害的功能！
但是呢，其实上面的装饰器非常不好。我写这样一个装饰器，只是为了帮助大家理解装饰器的执行过程和实现的结果。
请大家继续跟着我的思路：
我们先回顾一下：上面的目标函数login，当使用@decoration的意思是，login = decoration（login），我们把目标函数当作参数传给了decoration，然后decoration返回了一个inner给login。这时候login就已经不再是之前的login了，而是装饰函数里的inner函数的一个实例对象。当我们再执行login() 实际上执行了inner() ，原来的目标函数被装饰器的func保存了，在inner里面决定了什么时候执行func。
但是问题来了！
1 如果我们想给这个login目标函数传入参数：当调用login()的时候实际上调用的是inner(),如果我们想给login传参，但是却传参给了inner，inner又没有接收参数，这就会报错！！！
2 如果我们login函数有返回值，调用login()实际调用了inner()，inner内执行的func()才是执行的目标函数，func执行的时候又没有设置变量接收返回值，这样目标函数的返回值就丢了！！！
为了让我们的装饰器能够在各种各样的目标函数上使用，不论有没有参数，有没有返回值，我们都不会丢数据也不会报错，我们要对我们的装饰器进行修改，让它变得更加健壮！
一、通用的装饰器模板：
上代码！ 我们把装饰器函数简化一下，不要他那么复杂。请看下面的装饰器实例：
1 #装饰器函数 传入目标函数做参数
2 def decorator( func ):
3 #实际调用目标函数会发生调用inner，
4 # 所以我们让inner接收不定参数，我们再把不定参数传给目标函数func
5 #这样不论我们传入什么参数目标函数都能接收到！！
6 def inner( *args , **kwargs ):
7 print("装饰器函数中。。目标函数执行之前的操作！！")
8 # 我们设置一个变量接收目标函数的返回值，在inner结束的时候再把返回值返回去
9 # python不同于其他语言，即使我们没有编写func的返回值，也会默认返回None，所以这里不会报错
10 res = func( *args , **kwargs )
11 print("装饰器函数中。。目标函数执行之后的操作！！")
12 return res
13 return inner
14 #这样编写的装饰器，在外部看来，我们就可以传入参数给目标函数
15 # 同时也可以正常接收目标函数返回的参数!!!
16
17 @decorator #实际上会发生 destination = decorator( destination )
18 # 把目标函数传入装饰器函数返回了inner给destination
19 # 此后我们再调用destination 实际上调用了inner函数的一个对象
20 def destination( a ):
21 print( "目标函数接受到参数:%s"%a )
22 return "目标函数的返回值%s"%a
23
24 if __name__ == ‘__main__‘:
25 # 这里实际上调用了inner函数的对象，我们传入参数和接收返回值都没有问题了！
26 res = destination(10)
27 # 打印一下返回值！
28 print(res)
29 ‘‘‘
30 执行结果:
31 装饰器函数中。。目标函数执行之前的操作！！
32 目标函数接受到参数:10
33 装饰器函数中。。目标函数执行之后的操作！！
34 目标函数的返回值10
35 ‘‘‘
以上这个装饰器，实际上就可以作为一个通用版本的装饰器来使用了，它基本可以为各种各样的函数进行装饰。
我们传入目标函数的参数 在inner中设置不定参数去接收，又传给了func
func的返回值我们设置了变量res去接，又在inner结束时候返回了res
这样从外部看来，调用destination 只是单纯的被包装，并不知道里面这么复杂。
ok啦！！到这里呢，装饰器基本就介绍完了，下面来一点拓展！
二、多层装饰器嵌套：
1 #多层装饰器的嵌套
2
3 #外层装饰器函数
4 def decorator_out(func ):
5 def inner( *args , **kwargs ):
6 print("外装饰器前置操作。。。。。。。。。")
7 res = func( *args , **kwargs)
8 print("外装饰器后置操作..............")
9 return res
10 return inner
11
12 #内层装饰器函数
13 def decorator_in( func ):
14 def inner( *args , **kwargs ):
15 print("内装饰器前置操作。。。。。。。。。")
16 res = func( *args , **kwargs)
17 print("内装饰器后置操作..............")
18 return res
19 return inner
20
21 #目标函数被两个装饰器装饰
22 ‘‘‘
23 这里实际上会发生的情况是 login = decorator_out( decorator_in( destination ) )
24 先in装饰器装饰目标函数之后，把inner返回给destination,
25 然后 out装饰器 再对新的destination进行装饰，out里的func 存了 in装饰器装饰过的destination函数
26 又把inner返回给destination
27
28 这时候再执行目标函数destination() 实际上
29 1 执行out装饰器的inner() 执行到func的时候 执行in装饰器的inner
30 2 在内装饰器inner中执行func才是原来的目标函数
31 3 目标函数执行完跳出到in装饰器的inner
32 4 in装饰器函数inner执行完 相当于out装饰器的func执行完 跳到out装饰器的inner中
33 5 out装饰器执行结束，全部过程结束
34
35 ‘‘‘
36 @decorator_out
37 @decorator_in
38 def destination():
39 print("目标函数！")
40
41 if __name__ == ‘__main__‘:
42 destination()
43 ‘‘‘
44 执行结果：
45 外装饰器前置操作。。。。。。。。。
46 内装饰器前置操作。。。。。。。。。
47 目标函数！
48 内装饰器后置操作..............
49 外装饰器后置操作..............
50 ‘‘‘
当多层装饰器嵌套的时候，实际上先内层装饰器装饰目标函数
外层装饰器会对内层装饰器装饰的结果进行再装饰
不太好理解呀！！我很害怕好伙伴们理解不了我想表达的事情。如果是在不理解，不妨敲一敲代码，看一看注释，也许就懂了！！我也是懵懵懂懂学过来的~
三、可选择装饰器：
接下来我们再对装饰器来一个功能！
团队中一个人写了装饰器，大家一块用，之前的代码中，我写的装饰器，都是对目标函数执行之前和之后都有一些操作。如果某一个功能不需要之前或者之后的操作，只需要一个操作怎么办？？
这时候用到三层闭包嵌套。我们来一段代码理解一下
1 #我们传入一个flag决定是不是要执行目标函数之后的操作
2 def flagOperation( flag ):
3 def decorator(func):
4 def inner(*args , **kwargs):
5 print("装饰器前置操作。。。。。。。。")
6 res = func(*args , **kwargs)
7 if flag : #如果传入flag是真则执行后置操作 否则不执行
8 print("我是后置操作")
9 return res
10 return inner
11 return decorator
12
13
14
15 ‘‘‘
16 @flagOperation(True) 这里实际上先执行了flagOperation(True) ，返回了decorator装饰器，
17 并且把flag的值是True绑定给了他
18 这时候用带着flag的decorator 对目标函数进行装饰
19 相当于@decorator 带着一个flag
20 ‘‘‘
21 @flagOperation(True)
22 def destination():
23 print("目标函数！")
24
25 @flagOperation(False)
26 def desti2():
27 print("目标函数2")
28
29 if __name__ == ‘__main__‘:
30 destination()
31 ‘‘‘
32 结果：
33 装饰器前置操作。。。。。。。。
34 目标函数！
35 我是后置操作
36 ‘‘‘
37 desti2()
38 ‘‘‘
39 结果：
40 装饰器前置操作。。。。。。。。
41 目标函数2
42 ‘‘‘
ok啦，到目前位置，已经探讨了好多种牛逼的装饰器了，他们功能非常强大。学习要站在巨人的肩膀上，我么能完成更加厉害的功能。
对于理解有困难的好伙伴，只要努力的把我分享的 通用的装饰就可以了 其他可以不用理解。
四、类实现装饰器
如果对一个类名后跟() , 实际上会调用类内的__call__方法，所以我们把装饰器的逻辑写到__call__ 方法里就可以！
1 #类装配期 2 class Decorator(object): 3 def __init__(self,func): 4 self.func = func 5 def __call__(self): 6 print("类装饰器前操作") 7 self.func() 8 print("类装饰器后操作") 9 10 11 12 @Decorator 13 def login(): 14 print("login now") 15 16 if __name__ == ‘__main__‘: 17 login()
不装饰器到这里还有一些小问题没有解决。
目标函数是一个函数对象，它里面有自己的魔法属性 函数名__name__ 、说明文档__doc__等等。
如果我们目标函数被装饰器装饰之后，我们再调用 目标函数 destination.__name__ 会发现 本来 我们本来想要的 destination这个名字 打印出来确是 inner
这种情况确实不出乎意料，因为destination被装饰之后确实存的就是inner了，我们对destination的操作实际上都是对inner的操作
但是这带来了使用的不一致性。我们如果能把本来目标函数各种方法覆盖给inner函数就好了！
当然这是能实现的，我们可以用functools 包里的工具来实现。这里就不探讨了，再后续的博文中，我会再发表我学习functool模块的一些常用工具！
希望大家有所收获，我们共同进步！
欢迎其他朋友批评指正！
浅显易懂的谈一谈python中的装饰器！！