溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本文實例為大家分享了python修改裝飾器中參數(shù)的具體代碼，供大家參考，具體內(nèi)容如下
案例：
為分析程序內(nèi)哪些函數(shù)執(zhí)行時間開銷較大，我們需定義一個帶timeout參數(shù)的裝飾器
需求：
統(tǒng)計被裝飾函數(shù)的運行時間
時間大于timeout時，將此次函數(shù)調用記錄到log日志中
運行時可以修改timeout的值
如何解決這個問題？
#!/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)容，希望對大家的學習有所幫助，也希望大家多多支持億速云。
免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權內(nèi)容。
