Python利用装饰器计算程序运行时间的小demo

最近在运行一段程序时需要多次统计程序运行时间，并加以优化。想来也比较简单，但是每次都来一波时间统计时，就没必要次次都写一遍，自然就想到用很Pythonic的装饰器咯，一劳永逸嘛，下面直接上代码。

import time


def timer(func):
    def call_func(*args, **kwargs):
        print("计时开始")
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        total_time = end_time - start_time
        print("计时结束")
        print(f"程序用时{int(total_time // 60)}分{total_time % 60:.2f}秒")
    return call_func


if __name__ == '__main__':
    @timer
    def test(second):  # 测试函数
        time.sleep(second)

    test(3.4)

下面是运行结果

我们先实现一个通用的装饰器，并在func函数前后内置的time模块实现计时统计，并用python3.6推出的新特性f字符串（超级推荐，又快又直观，还简便）实现打印结果。
当我们需要统计某个程序或代码块的运行时长时（如上述test函数），在test函数上加上我们刚写的@timer装饰器就可以了。
当然我们需要复用这个模块，这时只需要将这个mytimer.py（以上代码块的名字，自己可以随意起别的）拷贝到我们的项目里面，导入该模块并在需要用到计时的地方用装饰器修饰就可以了，下面是一个实现的例子：

# 导入计时器装饰器
from mytimer import timer


@timer # 给函数加装饰器
def spark_demo(usetime):
    """
    假设我们要统计运行spark中spark_demo的总时长
    """
    import time
    time.sleep(usetime)
    pass


if __name__ == '__main__':
    spark_demo(74.4)

运行结果：

只要需要用到计时的地方加一个@timer，简简单单，舒舒服服就搞定了！这就是pythoner头发发量惊人的多的原因！！！