Python裝飾器(decorator)學習

G8bao7發表於2015-10-26
Python


點選(此處)摺疊或開啟

  1. #!/usr/bin/env python
  2. #coding=utf8
  3. import sys

  5. def deco(func):
  6.     print("before %s() called on %s." % (func, sys._getframe().f_code.co_name))
  7.     func()
  8.     print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  9.     return func

  11. def deco_at_one(func):
  12.     print("before %s() called on %s." % (func, sys._getframe().f_code.co_name))
  13.     func()
  14.     print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  15.     return func

  17. def deco_at_inline_noparams(func):
  18.     def _deco_inline():
  19.     print("before %s() called on %s." % (func, sys._getframe().f_code.co_name))
  20.     res = func()
  21.     print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  22.     return res
  23.     return _deco_inline

  25. def deco_at_inline_params_fix(func):
  26.     def _deco_inline_params_fix(a, b):
  27.     print("before %s() called on %s." % (func, sys._getframe().f_code.co_name))
  28.     res = func(a, b)
  29.     print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  30.     return res
  31.     return _deco_inline_params_fix

  33. def deco_at_inline_params_unfix(func):
  34.     def _deco_at_inline_params_unfix(*args, **kwargs):
  35.     print("before %s() called on %s." % (func, sys._getframe().f_code.co_name))
  36.     res = func(*args, **kwargs)
  37.     print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  38.     return res
  39.     return _deco_at_inline_params_unfix

  41. def deco_at_inline_params_unfix_withparams(dcargs):
  42.     def _deco_at_inline_params_unfix_wp(func):
  43.     def __deco_at_inline_params_unfix_wp(*args, **kwargs):
  44.      print("before %s() called on %s(%s)." % (func, sys._getframe().f_code.co_name, dcargs))
  45.      res = func(*args, **kwargs)
  46.      print("after %s() called on %s(%s)." % (func, sys._getframe().f_code.co_name, dcargs))
  47.      return res
  48.     return __deco_at_inline_params_unfix_wp
  49.     return _deco_at_inline_params_unfix_wp


  52. def deco_at_inline_params_unfix_wp_cls(dc_cls):
  53.     # dc_cls must implement static method function 'fn_sm_1' and 'fn_sm_2'
  54.     # see 'locker' module
  55.     def _deco_at_inline_params_unfix_wp_cls(func):
  56.     def __deco_at_inline_params_unfix_wp_cls(*args, **kwargs):
  57.      print("before %s() called on %s(%s)." % (func, sys._getframe().f_code.co_name, dc_cls))
  58.      dc_cls.fn_sm_1()
  59.      res = func(*args, **kwargs)
  60.      dc_cls.fn_sm_2()
  61.      print("after %s() called on %s." % (func, sys._getframe().f_code.co_name))
  62.      return res
  63.     return __deco_at_inline_params_unfix_wp_cls
  64.     return _deco_at_inline_params_unfix_wp_cls

  66. class Class_Deco():
  67.     def __init__(self):
  68.     print("INIT deco class." )

  70.     @staticmethod
  71.     def fn_sm_1():
  72.     print(" call %s." % (sys._getframe().f_code.co_name))
  73.     
  74.     @staticmethod
  75.     def fn_sm_2():
  76.     print(" call %s." % (sys._getframe().f_code.co_name))
  77.     
  78. '''
  79. -------------------------------------------------------------------------------------------------------------------
  80. '''
  81. def fn():
  82.     print("%s() called." % (sys._getframe().f_code.co_name))
  83.     return 'ok'

  85. def fn_deco():
  86.     print("%s() called." % (sys._getframe().f_code.co_name))
  87.     return 'ok'

  89. '''
  90. decorator function 'deco_at_one' called only one time before 'main'
  91. decorator function 'deco_at_one' not called when per time call 'fn_at_deco_one'
  92. '''
  93. @deco_at_one
  94. def fn_at_deco_one():
  95.     print("%s() called." % (sys._getframe().f_code.co_name))
  96.     return 'ok'

  98. '''
  99. decorator function 'deco_at_inline_noparams' called when per time call 'fn_at_deco_inline'
  100. '''
  101. @deco_at_inline_noparams
  102. def fn_at_deco_inline_noparams():
  103.     print("%s() called." % (sys._getframe().f_code.co_name))
  104.     return 'ok'

  106. '''
  107. params samed between '_deco_inline_params_fix' and 'fn_at_deco_inline'
  108. '''
  109. @deco_at_inline_params_fix
  110. def fn_at_deco_inline_params_fix(a,b):
  111.     print("%s() called." % (sys._getframe().f_code.co_name))
  112.     return a+ b

  114. '''
  115. 'deco_at_inline_params_unfix' any params
  116. '''
  117. @deco_at_inline_params_unfix
  118. def fn_at_deco_inline_params_unfix_2(a,b):
  119.     print("%s() called." % (sys._getframe().f_code.co_name))
  120.     return a+ b

  122. '''
  123. 'deco_at_inline_params_unfix' any params
  124. '''
  125. @deco_at_inline_params_unfix
  126. def fn_at_deco_inline_params_unfix_3(a,b,c):
  127.     print("%s() called." % (sys._getframe().f_code.co_name))
  128.     return a + b + c

  130. '''
  131. function 'deco_at_inline_params_unfix_withparams' can input params
  132. '''
  133. @deco_at_inline_params_unfix_withparams("wpa")
  134. def fn_at_deco_inline_params_unfix_wp_a(a,b,c):
  135.     print("%s() called." % (sys._getframe().f_code.co_name))
  136.     return a + b + c

  138. @deco_at_inline_params_unfix_withparams("wpb")
  139. def fn_at_deco_inline_params_unfix_wp_b(a,b,c):
  140.     print("%s() called." % (sys._getframe().f_code.co_name))
  141.     return a + b * c

  143. @deco_at_inline_params_unfix_wp_cls(Class_Deco)
  144. def fn_at_deco_inline_params_unfix_wp_cls():
  145.     print("%s() called." % (sys._getframe().f_code.co_name))
  146.     return 'ok'

  148. '''
  149. execute order by @deco_[1,2,3]
  150. @deco_1
  151. @deco_2
  152. @deco_3
  153. def fn():
  154.     pass

  156. '''
  157. @deco_at_inline_noparams
  158. @deco_at_inline_params_unfix_withparams("wpc")
  159. @deco_at_inline_params_unfix_wp_cls(Class_Deco)
  160. def fn_at_deco_inline_mix():
  161.     print("%s() called." % (sys._getframe().f_code.co_name))
  162.     return 'ok'

  164. if __name__ == '__main__':
  165.     lin = " - " * 30

  167.     print lin
  168.     fna = fn()

  170.     print lin
  171.     fnb = deco(fn_deco)

  173.     print lin
  174.     fnc = fn_at_deco_one()
  175.     fnc = fn_at_deco_one()

  177.     print lin
  178.     fnd = fn_at_deco_inline_noparams()
  179.     fnd = fn_at_deco_inline_noparams()

  181.     print lin
  182.     a=2
  183.     b=3
  184.     fne = fn_at_deco_inline_params_fix(a,b)
  185.     print a,b,fne

  187.     a=12
  188.     b=32
  189.     fne = fn_at_deco_inline_params_fix(a,b)
  190.     print a,b,fne

  192.     print lin
  193.     a=2
  194.     b=3
  195.     fnf = fn_at_deco_inline_params_unfix_2(a,b)
  196.     print a,b,fnf

  198.     a=2
  199.     b=3
  200.     c=5
  201.     fnf = fn_at_deco_inline_params_unfix_3(a,b,c)
  202.     print a,b,c,fnf

  204.     print lin
  205.     fng = fn_at_deco_inline_params_unfix_wp_a(a,b,c)
  206.     print a,b,c,fng
  207.     fng = fn_at_deco_inline_params_unfix_wp_b(a,b,c)
  208.     print a,b,c,fng

  210.     print lin
  211.     fng = fn_at_deco_inline_params_unfix_wp_cls()

  213.     print lin
  214.     fnh = fn_at_deco_inline_mix()

  216.     exit(0)


參考 : http://www.cnblogs.com/rhcad/archive/2011/12/21/2295507.html

其他參考資料

1. Python裝飾器學習 http://blog.csdn.net/thy38/article/details/4471421

2. Python裝飾器與面向切面程式設計 http://www.cnblogs.com/huxi/archive/2011/03/01/1967600.html

3. Python裝飾器的理解

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/26250550/viewspace-1816298/，如需轉載，請註明出處，否則將追究法律責任。

相關文章