首页 > 编程 > Python > 正文

python如何修改装饰器中参数

2020-01-04 15:34:49
字体:
来源:转载
供稿:网友

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

案例:

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

       需求:

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

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

    运行时可以修改timeout的值

如何解决这个问题?

  • 定义一个装饰器,计算函数执行时间,并与timeout比较,当大于timeout时候,通过logging模块打印出日志信息
  • 在包裹函数中添加一个函数,通过这个函数来修改timeout变量
  • 在python3中用nonlocal来声明嵌套作用域中的变量引用,在python2中可以通过把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武林网。


注:相关教程知识阅读请移步到python教程频道。
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表