python 装饰器实现输出每一行代码执行时间

た入场券 2023-07-23 06:58 39阅读 0赞

首先安装line\_profiler和wrapt包  
pip install line\_profiler  
![在这里插入图片描述][20200407171052501.png]  
pip install wrapt  
![在这里插入图片描述][20200407152603148.png]  
**装饰器：**

import wrapt
    from line_profiler import LineProfiler
    lp = LineProfiler()
     
    def lp_wrapper():
        @wrapt.decorator
        def wrapper(func, instance, args, kwargs):
            global lp
            lp_wrapper = lp(func)
            res = lp_wrapper(*args, **kwargs)
            lp.print_stats()
            return res
     
        return wrapper

放到需要查看的文件中即可，也可放到公共库里，这里就放到测试文件里：

import wrapt
    from line_profiler import LineProfiler
    
    lp = LineProfiler()
    
    
    def lp_wrapper():
        """ 显示函数调用时间 """
    
        @wrapt.decorator
        def wrapper(func, instance, args, kwargs):
            global lp
            lp_wrapper = lp(func)
            res = lp_wrapper(*args, **kwargs)
            lp.print_stats()
            return res
    
        return wrapper
    
    @lp_wrapper()
    def demo():
        # 执行的函数
        pass
    
    demo()

\*\*结果：\*\*此结果非上述代码的输出  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80Mjc1MjI0OA_size_16_color_FFFFFF_t_70]  
截图中最长的消耗了6.7秒  
这种方法是目前我所能知道的最方便的查看方法，只需要安装包写个装饰器即可，当函数被调用时即可查看被装饰的函数的详情。

[20200407171052501.png]: /images/20230528/0c1c5a49b0e74293a51721e5d8c987ea.png
[20200407152603148.png]: /images/20230528/047e20e6180f4b4aad5a0ba4335a86c5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80Mjc1MjI0OA_size_16_color_FFFFFF_t_70]: /images/20230528/83c48f1ef2ad4608aaf9b94697150653.png