使用python装饰器计算函数运行时间的实例
装饰器在python里面有很重要的作用,如果能够熟练使用,将会大大的提高工作效率
今天就来见识一下python装饰器,到底是怎么工作的。
本文主要是利用python装饰器计算函数运行时间
一些需要精确的计算函数运行了多久的程序,都可以采用这种方法
#coding:utf-8
importurllib2,re,time,random,os,datetime
importHTMLParser
importsys
reload(sys)
sys.setdefaultencoding('utf-8')
#计算时间函数
defprint_run_time(func):
defwrapper(*args,**kw):
local_time=time.time()
func(*args,**kw)
print'currentFunction[%s]runtimeis%.2f'%(func.__name__,time.time()-local_time)
returnwrapper
classtest:
def__init__(self):
self.url=''
#获取网页页面内容
#即装饰器不管参数有多少,都能使用
@print_run_time
defget_html(self,url):
headers={'User-Agent':'Mozilla/5.0(WindowsNT6.2;rv:16.0)Gecko/20100101Firefox/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())#转码:避免输出出现乱码
excepturllib2.HTTPError,e:
print(2,u"连接页面失败,错误原因:%s"%e.code)
returnNone
excepturllib2.URLError,e:
ifhasattr(e,'reason'):
print(2,u"连接页面失败,错误原因:%s"%e.reason)
returnNone
returnhtml
#在类的内部使用装饰器
@print_run_time
defrun(self):
self.url='http://www.baidu.com'
self.get_html(self.url)
print'end'
#在外面直接使用装饰器
@print_run_time
defget_current_dir(spath):
#spath=os.getcwd()
#spath=os.path.abspath(os.curdir)
forschildinos.listdir(spath):
schildpath=spath+'/'+schild
ifos.path.isdir(schildpath):
get_current_dir(schildpath)
else:
printschildpath
if__name__=='__main__':
my_test=test()
my_test.run()
spath=os.path.abspath('.')
get_current_dir(spath)
运行结果:
currentFunction[get_html]runtimeis0.29 end currentFunction[run]runtimeis0.29 05.python_study/03.decorator.py currentFunction[get_current_dir]runtimeis0.00
以上这篇使用python装饰器计算函数运行时间的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。