Python多重裝飾器

jitaolee

多重裝飾器，即多個(gè)裝飾器修飾同一個(gè)對(duì)象【實(shí)際上并非完全如此，且看下文詳解】

1.裝飾器無(wú)參數(shù)：

1. >>> def first(func):
   
2.     print '%s() was post to first()'%func.func_name
   
3.     def _first(*args,**kw):
   
4.         print 'Call the function %s() in _first().'%func.func_name
   
5.         return func(*args,**kw)
   
6.     return _first
   
7. 
   
8. 
   
9. >>> def second(func):
   
10.     print '%s() was post to second()'%func.func_name
    
11.     def _second(*args,**kw):
    
12.         print 'Call the function %s() in _second().'%func.func_name
    
13.         return func(*args,**kw)
    
14.     return _second
    
15. 
    
16. 
    
17. >>> @first
    
18. @second
    
19. def test():return 'hello world'
    
20. 
    
21. test() was post to second()
    
22. _second() was post to first()
    
23. >>> test()
    
24. Call the function _second() in _first().
    
25. Call the function test() in _second().
    
26. 'hello world'
    
27. >>>

復(fù)制代碼

實(shí)際上它是相當(dāng)于下面的代碼：

1. >>> def test():
   
2.     return 'hello world'
   
3. 
   
4. >>> test=second(test)
   
5. test() was post to second()
   
6. >>> test
   
7. <function _second at 0x000000000316D3C8>
   
8. >>> test=first(test)
   
9. _second() was post to first()
   
10. >>> test
    
11. <function _first at 0x000000000316D358>
    
12. >>> test()
    
13. Call the function _second() in _first().
    
14. Call the function test() in _second().
    
15. 'hello world'
    
16. >>>

復(fù)制代碼

2.裝飾器有參數(shù)：

1. >>> def first(printResult=False):
   
2.     def _first(func):
   
3.         print '%s() was post to _first()'%func.func_name
   
4.         def __first(*args,**kw):
   
5.             print 'Call the function %s() in __first().'%\
   
6.                   func.func_name
   
7.             if printResult:
   
8.                 print func(*args,**kw),'#print in __first().'
   
9.             else:
   
10.                 return func(*args,**kw)
    
11.         return __first
    
12.     return _first
    
13. 
    
14. >>> def second(printResult=False):
    
15.     def _second(func):
    
16.         print '%s() was post to _second()'%func.func_name
    
17.         def __second(*args,**kw):
    
18.             print 'Call the function %s() in __second().'%\
    
19.                   func.func_name
    
20.             if printResult:
    
21.                 print func(*args,**kw),'#print in __second().'
    
22.             else:
    
23.                 return func(*args,**kw)
    
24.         return __second
    
25.     return _second
    
26. 
    
27. >>> @first(True)
    
28. @second(True)
    
29. def test():
    
30.     return 'hello world'
    
31. 
    
32. test() was post to _second()
    
33. __second() was post to _first()
    
34. >>> test()
    
35. Call the function __second() in __first().
    
36. Call the function test() in __second().
    
37. hello world #print in __second().
    
38. None #print in __first().
    
39. >>>

復(fù)制代碼

如上，第35行輸出后調(diào)用__second()，而__second()中又調(diào)用了test()并print test()，而后返回__first()中繼續(xù)執(zhí)行print，而這個(gè)print語(yǔ)句print的內(nèi)容是__second()返回的None

它等同于：

1. >>> def test():
   
2.     return 'hello world'
   
3. 
   
4. >>> test=second(True)(test)
   
5. test() was post to _second()
   
6. >>>
   
7. >>> test
   
8. <function __second at 0x000000000316D2E8>
   
9. >>> test=first(True)(test)
   
10. __second() was post to _first()
    
11. >>> test
    
12. <function __first at 0x0000000003344C18>
    
13. >>>

復(fù)制代碼

3.多重裝飾器的應(yīng)用：

比如你是項(xiàng)目經(jīng)理，你要求每一個(gè)代碼塊都必須有參數(shù)檢查ArgsType和責(zé)任檢查ResponsibilityRegister，這樣就需要兩個(gè)裝飾器對(duì)此代碼塊進(jìn)行監(jiān)督。

1. #coding=utf-8
   
2. import os,sys,re
   
3. from collections import OrderedDict
   
4. 
   
5. def ArgsType(*argTypes,**kwTypes):
   
6.     u'''ArgsType(*argTypes,**kwTypes)
   
7.     options=[('opt_UseTypeOfDefaultValue',False)]
   
8. 
   
9.     以下為本函數(shù)相關(guān)的開(kāi)關(guān)，并非類(lèi)型檢驗(yàn)相關(guān)的關(guān)鍵字參數(shù)，所有options:
   
10.     opt_UseTypeOfDefaultValue=>bool:False,為T(mén)rue時(shí)，將對(duì)沒(méi)有指定類(lèi)型的帶默
    
11.                                認(rèn)值的參數(shù)使用其默認(rèn)值的類(lèi)型
    
12.     '''
    
13.     def _ArgsType(func):
    
14.         #確定所有的parameter name
    
15.         argNames=func.func_code.co_varnames[:func.func_code.co_argcount]
    
16.         #確定所有的default parameter
    
17.         defaults=func.func_defaults
    
18.         if defaults:
    
19.             defaults=dict(zip(argNames[-len(defaults):],defaults))
    
20.         else:defaults=None
    
21.         #將“參數(shù)類(lèi)型關(guān)鍵字參數(shù)”中的所有“options關(guān)鍵字參數(shù)”提出
    
22.         options=dict()
    
23.         for option,default in [('opt_UseTypeOfDefaultValue',False)]:
    
24.             options[option]=kwTypes.pop(option,default)
    
25.         #argTypes和kwTypes的總長(zhǎng)度應(yīng)該與argNames一致
    
26.         if len(argTypes)+len(kwTypes)>len(argNames):
    
27.             raise Exception('Too much types to check %s().'%func.func_name)
    
28.         #所有kwTypes中的鍵不能覆蓋在argTypes中已經(jīng)占用的names
    
29.         if not set(argNames[len(argTypes):]).issuperset(
    
30.             set(kwTypes.keys())):
    
31.             raise Exception('There is some key in kwTypes '+
    
32.                 'which is not in argNames.')
    
33.         #確定所有的參數(shù)應(yīng)該有的types
    
34.         types=OrderedDict()
    
35.         for name in argNames:types[name]=None
    
36.         if len(argTypes):
    
37.             for i in range(len(argTypes)):
    
38.                 name=argNames[i]
    
39.                 types[name]=argTypes[i]
    
40.         else:
    
41.             for name,t in kwTypes.items():
    
42.                 types[name]=t
    
43.         if len(kwTypes):
    
44.             for name,t in kwTypes.items():
    
45.                 types[name]=t
    
46.         #關(guān)于default parameter的type
    
47.         if options['opt_UseTypeOfDefaultValue']:
    
48.             for k,v in defaults.items():
    
49.                 #如果default parameter的type沒(méi)有另外指定，那么就使用
    
50.                 #default parameter的default value的type
    
51.                 if types[k]==None:
    
52.                     types[k]=type(v)
    
53.         def __ArgsType(*args,**kw):
    
54.             #order the args
    
55.             Args=OrderedDict()
    
56.             #init keys
    
57.             for name in argNames:Args[name]=None
    
58.             #init default values
    
59.             if defaults is not None:
    
60.                 for k,v in defaults.items():
    
61.                     Args[k]=v
    
62.             #fill in all args
    
63.             for i in range(len(args)):
    
64.                 Args[argNames[i]]=args[i]
    
65.             #fill in all keyword args
    
66.             for k,v in kw.items():
    
67.                 Args[k]=v
    
68.             #check if there is some None in the values
    
69.             if defaults==None:
    
70.                 for k in Args:
    
71.                     if Args[k]==None:
    
72.                         if defaults==None:
    
73.                             raise Exception(('%s() needs %r parameter, '+
    
74.                                 'which was not given')%(func.func_name,k))
    
75.                         else:
    
76.                            if not defaults.has_key(k):
    
77.                                 raise Exception(('Parameter %r of %s() is'+
    
78.                                     ' not a default parameter')%\
    
79.                                     (k,func.func_name))
    
80.             #check all types
    
81.             for k in Args:
    
82.                 if not isinstance(Args[k],types[k]):
    
83.                     raise TypeError(('Parameter %r of %s() must be '+
    
84.                         'a %r object, but you post: %r')%\
    
85.                         (k,func.func_name,types[k],Args[k]))
    
86.             return func(*args,**kw)
    
87.         return __ArgsType
    
88.     return _ArgsType
    
89. 
    
90. def ResponsibilityRegister(author):
    
91.     def _ResponsibilityRegister(func):
    
92.         def __ResponsibilityRegister(*args,**kw):
    
93.             try:
    
94.                 return func(*args,**kw)
    
95.             except Exception as e:
    
96.                 print ("Something is wrong, It's %s's responsibility."%\
    
97.                        author).center(80,'*')
    
98.                 raise e
    
99.         return __ResponsibilityRegister
    
100.     return _ResponsibilityRegister
     
101. 
     
102. @ResponsibilityRegister('Kate')
     
103. @ArgsType(str,int)
     
104. def left(Str,Len=1):
     
105.     return Str[:Len]
     
106. 
     
107. print 'Good calling:'
     
108. print left('hello world',8)
     
109. print 'Bad calling:'
     
110. print left(3,7)

復(fù)制代碼

這里沒(méi)有文檔，所以調(diào)用者不知道，使用了錯(cuò)誤的調(diào)用，導(dǎo)致出錯(cuò)，這是Kate的責(zé)任。

像上面這種，對(duì)代碼有兩種互不相干的檢驗(yàn)時(shí)，就可以使用多重裝飾器。

substr函數(shù)

MARK