使用python裝飾器計算函式執行時間
阿新 • • 發佈:2019-02-16
裝飾器在python裡面有很重要的作用, 如果能夠熟練使用,將會大大的提高工作效率
今天就來見識一下 python 裝飾器,到底是怎麼工作的。
本文主要是利用python裝飾器計算函式執行時間
一些需要精確的計算函式運行了多久的程式,都可以採用這種方法
#coding:utf-8 import urllib2,re,time,random,os,datetime import HTMLParser import sys reload(sys) sys.setdefaultencoding('utf-8') #計算時間函式 def print_run_time(func): def wrapper(*args, **kw): local_time = time.time() func(*args, **kw) print 'current Function [%s] run time is %.2f' % (func.__name__ ,time.time() - local_time) return wrapper class test: def __init__(self): self.url='' #獲取網頁頁面內容 #即裝飾器不管引數有多少,都能使用 @print_run_time def get_html(self,url): headers = {'User-Agent':'Mozilla/5.0 (Windows NT 6.2; rv:16.0) Gecko/20100101 Firefox/16.0'}#設定header req = urllib2.Request(url=url,headers=headers) try: html = urllib2.urlopen(req).read().decode('utf-8') html=HTMLParser.HTMLParser().unescape(html)#處理網頁內容, 可以將一些html型別的符號如" 轉換回雙引號 #html = html.decode('utf-8','replace').encode(sys.getfilesystemencoding())#轉碼:避免輸出出現亂碼 except urllib2.HTTPError,e: print(2,u"連線頁面失敗,錯誤原因: %s" % e.code) return None except urllib2.URLError,e: if hasattr(e,'reason'): print(2,u"連線頁面失敗,錯誤原因:%s" % e.reason) return None return html #在類的內部使用裝飾器 @print_run_time def run(self): self.url='http://www.baidu.com' self.get_html(self.url) print 'end' #在外面直接使用裝飾器 @print_run_time def get_current_dir(spath): #spath=os.getcwd() #spath=os.path.abspath(os.curdir) for schild in os.listdir(spath): schildpath=spath+'/'+schild if os.path.isdir(schildpath): get_current_dir(schildpath) else: print schildpath if __name__ == '__main__': my_test=test() my_test.run() spath=os.path.abspath('.') get_current_dir(spath)
執行結果:
current Function [get_html] run time is 0.29
end
current Function [run] run time is 0.29
05.python_study/03.decorator.py
current Function [get_current_dir] run time is 0.00