python如何修改装饰器中参数
本文实例为大家分享了python修改装饰器中参数的具体代码，供大家参考，具体内容如下
案例：
为分析程序内哪些函数执行时间开销较大，我们需定义一个带timeout参数的装饰器
需求：
统计被装饰函数的运行时间
时间大于timeout时，将此次函数调用记录到log日志中
运行时可以修改timeout的值
如何解决这个问题？
- 定义一个装饰器，计算函数执行时间，并与timeout比较，当大于timeout时候，通过logging模块打印出日志信息
- 在包裹函数中添加一个函数，通过这个函数来修改timeout变量
- 在python3中用nonlocal来声明嵌套作用域中的变量引用，在python2中可以通过把timeout参数变成列表，通过列表索引来进行改值
#!/usr/bin/python3
importtime
importlogging
fromrandomimportrandint
defrun_time(timeout):
"""
定义检查函数运行时间，并打印对应函数运行时间超出设定时间日志，并支持更改timeout
"""
#python2
#timeout=[timeout]
#真正包裹函数
defout_wrapper(func):
defwrapper(*args,**kwargs):
start_time=time.time()
result=func(*args,**kwargs)
used_time=time.time()-start_time
#对于超出timeout的函数进行日志打印
ifused_time>timeout:
msg='%s:%s>%s'%(func.__name__,used_time,timeout)
logging.warn(msg)
#python2
#ifused_time>timeout[0]:
#msg='%s:%s>%s'%(func.__name__,used_time,timeout[0])
#logging.warn(msg)
#returnresult
#设置timeout参数值
defset_timeout(value):
nonlocaltimeout
timeout=value
wrapper.set_timeout=set_timeout
#python2
#defset_timeout(value):
#timeout[0]=value
#wrapper.set_timeout=set_timeout
returnwrapper
returnout_wrapper
@run_time(1.5)
deffunc():
#随机有50%的几率程序沉睡1秒
whilerandint(0,1):
time.sleep(1)
print('func_run')
if__name__=="__main__":
for_inrange(10):
func()
print('_'*50)
#更改run_time装饰器中timeout参数
func.set_timeout(2)
for_inrange(10):
func()
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。