推荐两个非常实用的,Python装饰器
生活随笔
收集整理的這篇文章主要介紹了
推荐两个非常实用的,Python装饰器
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
1.超時(shí)函數(shù)
這個(gè)函數(shù)的作用在于可以給任意可能會(huì)hang住的函數(shù)添加超時(shí)功能,這個(gè)功能在編寫外部API調(diào)用 、網(wǎng)絡(luò)爬蟲、數(shù)據(jù)庫查詢的時(shí)候特別有用
timeout裝飾器的代碼如下:
import signal,functools #下面會(huì)用到的兩個(gè)庫 class TimeoutError(Exception): pass #定義一個(gè)Exception,后面超時(shí)拋出 def timeout(seconds, error_message = 'Function call timed out'):def decorated(func):def _handle_timeout(signum, frame):raise TimeoutError(error_message)def wrapper(*args, **kwargs):signal.signal(signal.SIGALRM, _handle_timeout)signal.alarm(seconds)try:result = func(*args, **kwargs)finally:signal.alarm(0)return resultreturn functools.wraps(func)(wrapper)return decorated使用:
@timeout(5) #限定下面的slowfunc函數(shù)如果在5s內(nèi)不返回就強(qiáng)制拋TimeoutError Exception結(jié)束 def slowfunc(sleep_time):import timetime.sleep(sleep_time) #這個(gè)函數(shù)就是休眠sleep_time秒 slowfunc(3) #sleep 3秒,正常返回 沒有異常 slowfunc(10) #被終止 ## 輸出 --------------------------------------------------------------------------- TimeoutError Traceback (most recent call last)2.Trace函數(shù)
有時(shí)候出于演示目的或者調(diào)試目的,我們需要程序運(yùn)行的時(shí)候打印出每一步的運(yùn)行順序 和調(diào)用邏輯。類似寫bash的時(shí)候的bash -x調(diào)試功能,然后Python解釋器并沒有 內(nèi)置這個(gè)時(shí)分有用的功能,那么我們就“自己動(dòng)手,豐衣足食”。
Trace裝飾器的代碼如下:
''' 學(xué)習(xí)中遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流QQ群:725638078 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(xué)習(xí)教程和PDF電子書! ''' import sys,os,linecache def trace(f):def globaltrace(frame, why, arg):if why == "call": return localtracereturn Nonedef localtrace(frame, why, arg):if why == "line":# record the file name and line number of every tracefilename = frame.f_code.co_filenamelineno = frame.f_linenobname = os.path.basename(filename)print("{}({}): {}".format( bname,lineno,linecache.getline(filename, lineno).strip('\r\n')),)return localtracedef _f(*args, **kwds):sys.settrace(globaltrace)result = f(*args, **kwds)sys.settrace(None)return resultreturn _f使用:
@trace def xxx():print (1)print (22)print (333)xxx() #調(diào)用 ## 輸出 <ipython-input-4-da50741ac84e>(3): print 1 # @trace 的輸出 1 <ipython-input-4-da50741ac84e>(4): print 22 # @trace 的輸出 22 <ipython-input-4-da50741ac84e>(5): print 333 # @trace 的輸出 333結(jié)尾給大家推薦一個(gè)非常好的學(xué)習(xí)教程,希望對你學(xué)習(xí)Python有幫助!
Python基礎(chǔ)入門教程推薦
Python爬蟲案例教程推薦
總結(jié)
以上是生活随笔為你收集整理的推荐两个非常实用的,Python装饰器的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python中异常的处理及断言,包括异常
- 下一篇: 5道python基础中while循环练习