本文實例為大家分享了python修改裝飾器中參數的具體代碼,供大家參考,具體內容如下
案例:
為分析程序內哪些函數執行時間開銷較大,我們需定義一個帶timeout參數的裝飾器
需求:
統計被裝飾函數的運行時間
時間大于timeout時,將此次函數調用記錄到log日志中
運行時可以修改timeout的值
如何解決這個問題?
#!/usr/bin/python3 import timeimport loggingfrom random import randint def run_time(timeout): """ 定義檢查函數運行時間,并打印對應函數運行時間超出設定時間日志,并支持更改timeout """ # python2 # timeout = [timeout] # 真正包裹函數 def out_wrapper(func): def wrapper(*args, **kwargs): start_time = time.time() result = func(*args, **kwargs) used_time = time.time() - start_time # 對于超出timeout的函數進行日志打印 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參數值 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參數 func.set_timeout(2) for _ in range(10): func()以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持VEVB武林網。
新聞熱點
疑難解答