【笔记】如何实现属性可修改的函数装饰器

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

　　1、统计被装饰函数单词调用运行时间

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

　　3、运行时可修改timeout的值

方法：为包裹增添一个函数，用来修改闭包中使用的自由变量。

　　　　在Python3中使用nonlocal访问嵌套作用域中的变量引用

# coding:utf8
from functools import wraps

import time
import logging


def warn(timeout):
    timeout = [timeout]
    def decorator(func):
        def wrapper(*args,**kargs):
            start = time.time()
            res = func(*args,**kargs)
            used = time.time() - start
            if used > timeout[0]:
                msg = '"%s": %s > %s' % (func.__name__,used,timeout[0])
                logging.warn(msg)
            return res
        def setTimeout(k):
            # Python3中有nonlocal来改变变量timeout的作用范围，Python2中使用可变列表来实现
            # nonlocal timeout
            # timeout = k
            timeout[0] = k
        wrapper.setTimeout = setTimeout
            
        return wrapper
    return decorator
    
from random import randint
@warn(1.5)
def test():
    print('in test')
    while randint(0,1):
        time.sleep(0.5)

for _ in range(30):
    test()
    
test.setTimeout(1)

for _ in range(30):
    test()