python如何修改装饰器中参数
更新时间:2018年03月20日 10:10:59 作者:北门吹雪
这篇文章主要为大家详细介绍了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()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
相关文章
使用Python将数组的元素导出到变量中(unpacking)
最近工作中遇到一个问题,需要利用Python将数组(list)或元组(tuple)中的元素导出到N个变量中,现在将我实现的方法分享给大家,有需要的朋友们可以参考借鉴,下面来一起看看吧。2016-10-10Python 对象序列化与反序列化之pickle json详细解析
我们知道在Python中,一切皆为对象,实例是对象,类是对象,元类也是对象。本文正是要聊聊如何将这些对象有效地保存起来,以供后续使用2021-09-09
最新评论