Python編程（bbb四）：兩個實用的Python的裝飾器

超時函數

這個函數的作用在於可以給任意可能會hang住的函數添加超時功能，這個功能在編寫外部API調用 、網路爬蟲、資料庫查詢的時候特別有用

timeout裝飾器的代碼如下：

import signal,functools #下面會用到的兩個庫 nclass TimeoutError(Exception): pass #定義一個Exception，後面超時拋出 nndef timeout(seconds, error_message = Function call timed out):n def decorated(func):n def _handle_timeout(signum, frame):n raise TimeoutError(error_message)n def wrapper(*args, **kwargs):n signal.signal(signal.SIGALRM, _handle_timeout)n signal.alarm(seconds)n try:n result = func(*args, **kwargs)n finally:n signal.alarm(0)n return resultn return functools.wraps(func)(wrapper)n return decoratedn

使用：

@timeout(5) #限定下面的slowfunc函數如果在5s內不返回就強制拋TimeoutError Exception結束 ndef slowfunc(sleep_time):n import timen time.sleep(sleep_time) #這個函數就是休眠sleep_time秒 nnslowfunc(3) #sleep 3秒，正常返回 沒有異常 nnnslowfunc(10) #被終止 nn## 輸出 n---------------------------------------------------------------------------nTimeoutError Traceback (most recent call last)n

Trace函數

有時候出於演示目的或者調試目的，我們需要程序運行的時候列印出每一步的運行順序 和調用邏輯。類似寫bash的時候的bash -x調試功能，然後Python解釋器並沒有 內置這個時分有用的功能，那麼我們就「自己動手，豐衣足食」。

Trace裝飾器的代碼如下：

import sys,os,linecachendef trace(f):n def globaltrace(frame, why, arg):n if why == "call": return localtracen return Nonen def localtrace(frame, why, arg):n if why == "line":n # record the file name and line number of every trace n filename = frame.f_code.co_filenamen lineno = frame.f_linenon bname = os.path.basename(filename)n print "{}({}): {}".format( bname,n lineno,n linecache.getline(filename, lineno).strip(rn)),n return localtracen def _f(*args, **kwds):n sys.settrace(globaltrace)n result = f(*args, **kwds)n sys.settrace(None)n return resultn return _fn

使用：

@tracendef xxx():n print 1n print 22n print 333nnxxx() #調用 nn## 輸出 n<ipython-input-4-da50741ac84e>(3): print 1 # @trace 的輸出 n1n<ipython-input-4-da50741ac84e>(4): print 22 # @trace 的輸出 n22n<ipython-input-4-da50741ac84e>(5): print 333 # @trace 的輸出 n333n

推薦閱讀：