9-4修改属性的装饰器

# -*- coding:utf-8 -*- ''' 为分析程序内哪些函数执行时间开销较大,定义一个timeout参数的函数装饰器.装饰功能: 1.统计被装饰函数单词调用运行时间. 2.时间大于参数timeout的,将此次函数调用记录到log日志中. 3.运行时可修改timeout的值. 解决方案: 为包裹函数增添一个函数,用来修改闭包中使用的自由变量. 在python3中:使用nonlocal访问嵌套作用域中的变量引用 ''' from functools import wraps import time import logging from random import randint def warn(timeout): def decorator(func): @wraps(func) def wrapper(*args, **kwargs): start = time.time() res = func(*args, **kwargs) used = time.time() - start if used > timeout: msg = '"%s": %s > %s' % (func.__name__, used, timeout) logging.warn(msg) return res def setTimeout(k): # timeout = k # 这样是不允许的,因为timeout是在函数wrapper闭包内使用的自由变量 # 此句只会在setTimeout()创建一个属于本函数的本地变量 nonlocal timeout # 类似于global的关键字,使用在嵌套作用域下 timeout = k wrapper.setTimeout = setTimeout # 将setTimeout()函数作为wrapper的属性 return wrapper return decorator # 因为python-2不支持nolocal关键字,所以做以下处理 def warn1(timeout): timeout = [timeout] # attr -> [attr,] def decorator(func): @wraps(func) def wrapper(*args, **kwargs): start = time.time() res = func(*args, **kwargs) used = time.time() - start if used > timeout[0]: msg = '"%s": %s > %s' % (func.__name__, used, timeout[0]) # timeout -> timeout[0] logging.warn(msg) return res def setTimeout(k): timeout[0] = k # timeout -> timeout[0] wrapper.setTimeout = setTimeout # 将setTimeout()函数作为wrapper的属性 return wrapper return decorator @warn1(1.5) def test(): print('IN test') while randint(0, 1): time.sleep(0.5) if __name__ == '__main__': for _ in range(30): test() test.setTimeout(1) for _ in range(30): test()