python_如何修改装饰器中参数？

Posted 2020-10-01

案例：

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

       需求：

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

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

　　　　运行时可以修改timeout的值

如何解决这个问题？

1. 定义一个装饰器，计算函数执行时间，并与timeout比较，当大于timeout时候，通过logging模块打印出日志信息
2. 在包裹函数中添加一个函数，通过这个函数来修改timeout变量
3. 在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()