1. 程式人生 > >使用python裝飾器計算函式執行時間

使用python裝飾器計算函式執行時間

裝飾器在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