修行Python個把星期,終於有點成果了,呵呵,一個利用metaclass實現的aop。
其實python這類非常動態語言要實現AOP是很容易的,所以首先我們要來先定義一個metaclass
然後我們要在__new__()這個metaclass 的時候動態植入方法到要調用地方法的前後。
具體代碼如下:
1
2__author__="alex"
3__date__ ="$2008-12-5 23:54:11$"
4__name__="pyaop"
5
6'''
7這個metaclass是實現AOP的基礎
8'''
9class pyaop(type):
10 '''
11 這個空方法是用來將後面的beforeop和afterop初始化成函數引用
12 '''
13 def nop(self):
14 pass
15 '''
16 下面這兩個變數是類變數,也就是存放我們要植入的兩個函數的地址的變數
17 '''
18 beforeop=nop
19 afterop=nop
20 '''
21 設定前後兩個植入函數的類函數
22 '''
23 @classmethod
24 def setbefore(self,func):
25 pyaop.beforeop=func
26 @classmethod
27 def setafter(self,func):
28 pyaop.afterop=func
29 '''
30 初始化metaclass的函數,這個函數最重要的就是第四個參數,dict通過這個參數我們可以修改類的屬性(方法)
31 '''
32 def __new__(mcl,name,bases,dict):
33 from types import FunctionType #載入類型模組的FunctionType
34 obj=object() #定義一個Null 物件的變數
35 '''
36 這個就是要植入的方法,func參數就是我們要調用的函數
37 '''
38 def aop(func):
39 '''
40 我們用這個函數來代替將要調用的函數
41 '''
42 def wrapper(*args, **kwds):
43 pyaop.beforeop(obj) #調用前置函數
44 value = func(*args, **kwds) #調用本來要調用的函數
45 pyaop.afterop(obj) #調用後置函數
46 return value #返回
47 return wrapper
48 #在類的成員列表中尋找即將調用的函數
49 for attr, value in dict.iteritems():
50 if isinstance(value, FunctionType):
51 dict[attr] = aop(value) #找到後用aop這個函數替換之
52 obj=super(pyaop, mcl).__new__(mcl, name, bases, dict) #調用父類的__new__()建立self
53 return obj
54
使用的時候,如果我們要攔截一個類A的方法調用,就這樣子:
1class A(object):
2 __metaclass__ = pyaop
3 def foo(self):
4 total = 0
5 for i in range(100000):
6 total = total+1
7 print total
8
9 def foo2(self):
10 from time import sleep
11 total = 0
12 for i in range(100000):
13 total = total+1
14 sleep(0.0001)
15 print total
最後我們只需要:
1def beforep(self):
2 print('before')
3def afterp(self):
4 print('after')
5
6if __name__ == "__main__":
7 pyaop.setbefore(beforep)
8 pyaop.setafter(afterp)
9 a=A()
10 a.foo()
11 a.foo2()
這樣子在執行代碼的時候就得到了如下結果
before
100000
after
before
100000
after
這段代碼耗時一天才調通(汗顏),是基於園裡另外一貼 http://www.cnblogs.com/cavingdeep/archive/2006/08/22/483056.html
並感謝python討論群69828975裡的大蝦Seewind(310380)給予的無私協助
歡迎各位python大牛給予斧正