快捷導(dǎo)航

兩個(gè)很實(shí)用的Python裝飾器詳解

更新時(shí)間：2021年11月23日 17:06:43 作者：Python探索牛

這篇文章主要為大家介紹了Python的裝飾器，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助，希望能夠給你帶來幫助

1.超時(shí)函數(shù)

這個(gè)函數(shù)的作用在于可以給任意可能會hang住的函數(shù)添加超時(shí)功能，這個(gè)功能在編寫外部API調(diào)用、網(wǎng)絡(luò)爬蟲、數(shù)據(jù)庫查詢的時(shí)候特別有用

timeout裝飾器的代碼如下：

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

使用：

@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裝飾器的代碼如下：

'''
學(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

使用：

@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