1、python裝飾器

?

?

最基礎(chǔ)的函數(shù)

            1 def sum1():
2      sum = 1 + 2
3      print(sum)
4 sum1()
          

查看函數(shù)執(zhí)行用了多長(zhǎng)時(shí)間，寫了幾句代碼插進(jìn)去了：

             1 import time
 2 
 3 def sum1():
 4     start = time.clock()
 5     sum = 1+2
 6     print(sum)
 7     end = time.clock()
 8     print("time used:",end - start)
 9 
10 sum1()
          

?

可是隨著繼續(xù)翻看，對(duì)越來越多的函數(shù)感興趣了，都想看下他們的運(yùn)行時(shí)間如何，難道要一個(gè)一個(gè)的去改函數(shù)嗎？當(dāng)然不是！我們可以考慮重新定義一個(gè)函數(shù)timeit，將sum1的引用傳遞給他，然后在timeit中調(diào)用sum1并進(jìn)行計(jì)時(shí)，這樣，我們就達(dá)到了不改動(dòng)sum1定義的目的，而且，不論看了多少個(gè)函數(shù)，我們都不用去修改函數(shù)定義了！

            import time

def sum1():
    sum = 1+ 2
    print (sum)

def timeit(func):
    start = time.clock()
    func()
    end =time.clock()
    print("time used:", end - start)

timeit(sum1)
          

咂一看，沒啥問題，可以運(yùn)行！但是還是修改了一部分代碼，把sum1() 改成了timeit(sum1)。這樣的話，如果sum1在N處都被調(diào)用了，你就不得不去修改這N處的代碼。所以，我們就需要楊sum1()具有和timeit(sum1)一樣的效果，于是將timeit賦值給sum1。可是timeit是有參數(shù)的，所以需要找個(gè)方法去統(tǒng)一參數(shù)，將timeit(sum1)的返回值（計(jì)算運(yùn)行時(shí)間的函數(shù)）賦值給sum1。

             1 import time
 2 
 3 def sum1():
 4     sum = 1+ 2
 5     print (sum)
 6 
 7 def timeit(func):
 8     def test():
 9         start = time.clock()
10         func()
11         end =time.clock()
12         print("time used:", end - start)
13     return test
14 
15 sum1 = timeit(sum1)
16 sum1()
          

這樣一個(gè)簡(jiǎn)易的裝飾器就做好了，我們只需要在定義sum1以后調(diào)用sum1之前，加上sum1= timeit(sum1)，就可以達(dá)到計(jì)時(shí)的目的，這也就是裝飾器的概念，看起來像是sum1被timeit裝飾了！Python于是提供了一個(gè)語法糖來降低字符輸入量。

             1 import time
 2   
 3 def timeit(func):
 4      def test():
 5          start = time.clock()
 6          func()
 7          end =time.clock()
 8          print("time used:", end - start)
 9      return test
10  
11 @timeit
12 def sum1():
13      sum = 1+ 2
14      print (sum)
15 
16  sum1()
          

重點(diǎn)關(guān)注第11行的@timeit，在定義上加上這一行與另外寫sum1 = timeit(sum1)完全等價(jià)。