兩個很實用的Python裝飾器詳解
1.超時函數(shù)
這個函數(shù)的作用在于可以給任意可能會hang住的函數(shù)添加超時功能，這個功能在編寫外部API調(diào)用 、網(wǎng)絡(luò)爬蟲、數(shù)據(jù)庫查詢的時候特別有用
timeout裝飾器的代碼如下：
import signal,functools #下面會用到的兩個庫 class TimeoutError(Exception): pass #定義一個Exception，后面超時拋出 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 result return functools.wraps(func)(wrapper) return decorated
使用：
@timeout(5) #限定下面的slowfunc函數(shù)如果在5s內(nèi)不返回就強制拋TimeoutError Exception結(jié)束 def slowfunc(sleep_time): import time time.sleep(sleep_time) #這個函數(shù)就是休眠sleep_time秒 slowfunc(3) #sleep 3秒，正常返回 沒有異常 slowfunc(10) #被終止
## 輸出
---------------------------------------------------------------------------
TimeoutError Traceback (most recent call last)
2.Trace函數(shù)
有時候出于演示目的或者調(diào)試目的，我們需要程序運行的時候打印出每一步的運行順序 和調(diào)用邏輯。類似寫bash的時候的bash -x調(diào)試功能，然后Python解釋器并沒有 內(nèi)置這個時分有用的功能，那么我們就“自己動手，豐衣足食”。
Trace裝飾器的代碼如下：
'''
學(xué)習(xí)中遇到問題沒人解答？小編創(chuàng)建了一個Python學(xué)習(xí)交流群：725638078
尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學(xué)習(xí)教程和PDF電子書！
'''
import sys,os,linecache
def trace(f):
def globaltrace(frame, why, arg):
if why == "call": return localtrace
return None
def localtrace(frame, why, arg):
if why == "line":
# record the file name and line number of every trace
filename = frame.f_code.co_filename
lineno = frame.f_lineno
bname = os.path.basename(filename)
print("{}({}): {}".format( bname,
lineno,
linecache.getline(filename, lineno).strip('\r\n')),)
return localtrace
def _f(*args, **kwds):
sys.settrace(globaltrace)
result = f(*args, **kwds)
sys.settrace(None)
return result
return _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é)
本篇文章就到這里了，希望能夠給你帶來幫助，也希望您能夠多多關(guān)注本站的更多內(nèi)容!
版權(quán)聲明：本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場，如有內(nèi)容涉嫌侵權(quán)，請聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信