python 實現計時器，統計執行時長

ponponon發表於2022-02-21
Python

我們用兩種方式來實現計時器。

  • 修飾器實現計時器
  • 生成器實現計時器

修飾器實現計時器

import time


def timer_dec(func):
    def inner(*args, **kwargs):
        start = time.time()
        ret_val = func(*args, **kwargs)
        end = time.time()
        print(f'pay time is {end-start} s')
        return ret_val
    return inner


@timer_dec
def haha():
    print('笑死我了')
    time.sleep(0.01)


haha()

輸出結果

笑死我了
pay time is 0.010136842727661133 s

生成器實現計時器

使用生成器的方式實現，需要藉助 Python 內建庫 contextlib 幫助。

不是說必須藉助 contextlib，而是用 contextlib 可以更加優雅
可以參考：淺淡python中with的用法，上下文管理器
from contextlib import contextmanager

def haha():
    print('笑死我了')
    time.sleep(0.01)


@contextmanager
def timer_gen():
    start = time.time()
    yield
    end = time.time()
    print(f'pay time is {end-start} s')


with timer_gen():
    haha()

輸出結果

笑死我了
pay time is 0.010103464126586914 s

相關文章