兩個很實用的Python裝飾器詳解
1.超時函數(shù)
這個函數(shù)的作用在于可以給任意可能會hang住的函數(shù)添加超時功能,這個功能在編寫外部API調用 、網(wǎng)絡爬蟲、數(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內不返回就強制拋TimeoutError Exception結束 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ù)
有時候出于演示目的或者調試目的,我們需要程序運行的時候打印出每一步的運行順序 和調用邏輯。類似寫bash的時候的bash -x調試功能,然后Python解釋器并沒有 內置這個時分有用的功能,那么我們就“自己動手,豐衣足食”。
Trace裝飾器的代碼如下:
''' 學習中遇到問題沒人解答?小編創(chuàng)建了一個Python學習交流群:725638078 尋找有志同道合的小伙伴,互幫互助,群里還有不錯的視頻學習教程和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() #調用
## 輸出
<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
總結
本篇文章就到這里了,希望能夠給你帶來幫助,也希望您能夠多多關注腳本之家的更多內容!
相關文章
Python實現(xiàn)的合并兩個有序數(shù)組算法示例
這篇文章主要介紹了Python實現(xiàn)的合并兩個有序數(shù)組算法,涉及Python針對數(shù)組的遍歷、計算、追加等相關操作技巧,需要的朋友可以參考下2019-03-03pandas.DataFrame寫入數(shù)據(jù)庫的實現(xiàn)方式
這篇文章主要介紹了pandas.DataFrame寫入數(shù)據(jù)庫的實現(xiàn)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08Python函數(shù)式編程之返回函數(shù)實例詳解
函數(shù)式編程的一個特點就是,允許把函數(shù)本身作為參數(shù)傳入另一個函數(shù),還允許返回一個函數(shù),下面這篇文章主要給大家介紹了關于Python函數(shù)式編程之返回函數(shù)的相關資料,需要的朋友可以參考下2022-09-09使用python監(jiān)測網(wǎng)絡連接和網(wǎng)速的實現(xiàn)代碼
在我們日常生活中網(wǎng)絡連接和網(wǎng)速在工作中非常重要,本文將介紹如何使用Python程序來監(jiān)測互聯(lián)網(wǎng)連接的速度和中斷情況,并通過代碼示例講解的非常詳細,需要的朋友可以參考下2024-03-03