python如何修改裝飾器中參數(shù)

更新時間：2018年03月20日 10:10:59 作者：北門吹雪

這篇文章主要為大家詳細介紹了python如何修改裝飾器中參數(shù)，具有一定的參考價值，感興趣的小伙伴們可以參考一下

本文實例為大家分享了python修改裝飾器中參數(shù)的具體代碼，供大家參考，具體內(nèi)容如下

案例：

為分析程序內(nèi)哪些函數(shù)執(zhí)行時間開銷較大，我們需定義一個帶timeout參數(shù)的裝飾器

需求：

　　　　統(tǒng)計被裝飾函數(shù)的運行時間

　　　　時間大于timeout時，將此次函數(shù)調(diào)用記錄到log日志中

　　　　運行時可以修改timeout的值

如何解決這個問題？

定義一個裝飾器，計算函數(shù)執(zhí)行時間，并與timeout比較，當大于timeout時候，通過logging模塊打印出日志信息
在包裹函數(shù)中添加一個函數(shù)，通過這個函數(shù)來修改timeout變量
在python3中用nonlocal來聲明嵌套作用域中的變量引用，在python2中可以通過把timeout參數(shù)變成列表，通過列表索引來進行改值

#!/usr/bin/python3
 
import time
import logging
from random import randint
 
 
def run_time(timeout):
  """
  定義檢查函數(shù)運行時間，并打印對應函數(shù)運行時間超出設定時間日志，并支持更改timeout
  """
 
  # python2
  # timeout = [timeout]
   
  # 真正包裹函數(shù)
  def out_wrapper(func):
    def wrapper(*args, **kwargs):
       
      start_time = time.time()
      result = func(*args, **kwargs)
      used_time = time.time() - start_time
       
      # 對于超出timeout的函數(shù)進行日志打印
      if used_time > timeout:
        msg = '%s: %s > %s' % (func.__name__, used_time, timeout)
        logging.warn(msg)
       
      # python2
      # if used_time > timeout[0]:
      #   msg = '%s: %s > %s' % (func.__name__, used_time, timeout[0])
      #   logging.warn(msg)
      # return result
     
    # 設置timeout參數(shù)值
    def set_timeout(value):
      nonlocal timeout
      timeout = value
    wrapper.set_timeout = set_timeout
     
    # python2
    # def set_timeout(value):
    #   timeout[0] = value
    # wrapper.set_timeout = set_timeout
     
    return wrapper
  return out_wrapper
 
 
@run_time(1.5)
def func():
  # 隨機有50%的幾率程序沉睡1秒
  while randint(0, 1):
    time.sleep(1)
  print('func_run')
 
if __name__ == "__main__":
  for _ in range(10):
    func()
     
  print('_'*50)
   
  # 更改run_time裝飾器中timeout參數(shù)
  func.set_timeout(2)
  for _ in range(10):
    func()

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: