超時函數

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

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函數如果在5s內不返回就強制拋TimeoutError Exception結束 def slowfunc(sleep_time): import time time.sleep(sleep_time) #這個函數就是休眠sleep_time秒 slowfunc(3) #sleep 3秒，正常返回 沒有異常 slowfunc(10) #被終止 ## 輸出 --------------------------------------------------------------------------- TimeoutError Traceback (most recent call last)

Trace函數

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

Trace裝飾器的代碼如下：

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


作者：auxten
鏈接：https://zhuanlan.zhihu.com/p/20175869

總結

以上是生活随笔為你收集整理的两个实用的Python的装饰器的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。