兩個實用的Python的裝飾器

超時函數

　　這個函數的做用在於能夠給任意可能會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)),
    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