1.超時(shí)函數(shù)
這個(gè)函數(shù)的作用在于可以給任意可能會hang住的函數(shù)添加超時(shí)功能,這個(gè)功能在編寫外部API調(diào)用 、網(wǎng)絡(luò)爬蟲、數(shù)據(jù)庫查詢的時(shí)候特別有用
timeout裝飾器的代碼如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
import signal,functools #下面會用到的兩個(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 result return functools.wraps(func)(wrapper) return decorated |
使用:
1
2
3
4
5
6
|
@timeout ( 5 ) #限定下面的slowfunc函數(shù)如果在5s內(nèi)不返回就強(qiáng)制拋TimeoutError Exception結(jié)束 def slowfunc(sleep_time): import time time.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í)分有用的功能,那么我們就“自己動手,豐衣足食”。
Trace裝飾器的代碼如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
''' 學(xué)習(xí)中遇到問題沒人解答?小編創(chuàng)建了一個(gè)Python學(xué)習(xí)交流群:725638078 尋找有志同道合的小伙伴,互幫互助,群里還有不錯(cuò)的視頻學(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 |
使用:
1
2
3
4
5
6
|
@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)注服務(wù)器之家的更多內(nèi)容!
原文鏈接:https://www.cnblogs.com/djdjdj123/p/15563756.html