快捷導(dǎo)航

python實現(xiàn)Decorator模式實例代碼

更新時間：2018年02月09日 15:43:02 作者：JasonLi-九黎

這篇文章主要介紹了python實現(xiàn)Decorator模式實例代碼，簡單介紹了裝飾器的含義和語法，分享了相關(guān)實例，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下

本文研究的主要是python實現(xiàn)Decorator模式，具體介紹如下。

一般來說，裝飾器是一個函數(shù)，接受一個函數(shù)（或者類）作為參數(shù)，返回值也是也是一個函數(shù)（或者類）。首先來看一個簡單的例子：

# -*- coding: utf-8 -*-
def log_cost_time(func):
  def wrapped(*args, **kwargs):
    import time
    begin = time.time()
    try:
      return func(*args, **kwargs)
    finally:
      print 'func %s cost %s' % (func.__name__, time.time() - begin)
  return wrapped
 
@log_cost_time
def complex_func(num):
  ret = 0
  for i in xrange(num):
    ret += i * i
  return ret
#complex_func = log_cost_time(complex_func)
 
if __name__ == '__main__':
  print complex_func(100000)
 
code snippet 0

代碼中，函數(shù)log_cost_time就是一個裝飾器，其作用也很簡單，打印被裝飾函數(shù)運(yùn)行時間。

裝飾器的語法如下：

@dec

def func():pass

本質(zhì)上等同于： func = dec(func) 。

在上面的代碼（code snippet 0）中，把line12注釋掉，然后把line18的注釋去掉，是一樣的效果。另外staticmethod和classmethod是兩個我們經(jīng)常在代碼中用到的裝飾器，如果對pyc反編譯，得到的代碼一般也都是 func = staticmthod(func)這種模式。當(dāng)然，@符號的形式更受歡迎些，至少可以少拼寫一次函數(shù)名。

實例代碼

#-*-coding:utf-8-*-


'''
意圖：動態(tài)地給一個對象添加一些額外的職責(zé)。比通過生成子類更為靈活
'''
from abc import ABCMeta

class Component():
  __metaclass__ = ABCMeta
  def __init__(self):
    pass
  def operation(self):
    pass
  
class ConcreteComponent(Component):
  def operation(self):
    print 'ConcreteComponent operation...'

class Decorator(Component):
  def __init__(self, comp):
    self._comp = comp
  def operation(self):
    pass

class ConcreteDecorator(Decorator):
  def operation(self):
    self._comp.operation()
    self.addedBehavior()
  def addedBehavior(self):
    print 'ConcreteDecorator addedBehavior...' 
       
if __name__ == "__main__":
   comp = ConcreteComponent()
   dec = ConcreteDecorator(comp)
   dec.operation()

結(jié)果