python如何修改装饰器中参数

本文实例为大家分享了python修改装饰器中参数的具体代码，供大家参考，具体内容如下

案例：

       为分析程序内哪些函数执行时间开销较大，我们需定义一个带timeout参数的装饰器

       需求：

统计被装饰函数的运行时间

时间大于timeout时，将此次函数调用记录到log日志中

运行时可以修改timeout的值

如何解决这个问题？

• 定义一个装饰器，计算函数执行时间，并与timeout比较，当大于timeout时候，通过logging模块打印出日志信息

• 在包裹函数中添加一个函数，通过这个函数来修改timeout变量

• 在python3中用nonlocal来声明嵌套作用域中的变量引用，在python2中可以通过把timeout参数变成列表，通过列表索引来进行改值

#!/usr/bin/python3

import time
import logging
from 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()

来源：http://www.cnblogs.com/2bjiujiu/p/7293912.html