python 寫一個計算執行時間的裝飾器

標籤：

面試題之一。

寫一個裝飾器wraps，它接收一個參數t，如果函數執行時間超過1秒，輸出“bad”，否則輸出“goods”。

首先，計算函數的執行時間：

import timestart = time.clock()for i in range(1000000):    passend = time.clock()print "cost time = %f s" % (end-start)

結果：

>>> cost time = 0.092749 s>>> 

然後：

import timedef warps(t):    def deco(func):        def _deco(*args,**kwargs):            start = time.clock()            func(*args,**kwargs)            end = time.clock()            if end - start > t:                print 'bad'            else:                print 'goods'        return _deco    return deco@warps(10)def myfunc(*args,**kwargs):    for i in range(100000):        pass    myfunc()  

結果：

>>> goods>>> 

參考文檔：九步學習裝飾器

著作權聲明：本文為博主原創文章，未經博主允許不得轉載。

python 寫一個計算執行時間的裝飾器