首页 > 代码库 > python_如何修改装饰器中参数?

python_如何修改装饰器中参数?

案例:

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

       需求:

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

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

    运行时可以修改timeout的值

如何解决这个问题?

  1. 定义一个装饰器,计算函数执行时间,并与timeout比较,当大于timeout时候,通过logging模块打印出日志信息
  2. 在包裹函数中添加一个函数,通过这个函数来修改timeout变量
  3. 在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_如何修改装饰器中参数?