测量时间的 Python 上下文管理器

Question

我正在努力制作一段代码，允许测量在“with”语句中花费的时间并将测量的时间( float )分配给“with”语句中提供的变量。

import time

class catchtime:
    def __enter__(self):
        self.t = time.clock()
        return 1

    def __exit__(self, type, value, traceback):
        return time.clock() - self.t

with catchtime() as t:
    pass

此代码留下 t=1 而不是 clock() 调用之间的区别。如何解决这个问题？我需要一种从退出方法中分配新值的方法。

PEP 343 describes更详细地介绍了 contect manager 是如何工作的，但我不明白其中的大部分内容。

Answer 1

这是一个使用 contextmanager 的例子

from time import perf_counter
from contextlib import contextmanager

@contextmanager
def catchtime() -> float:
    start = perf_counter()
    yield lambda: perf_counter() - start


with catchtime() as t:
    import time
    time.sleep(1)

print(f"Execution time: {t():.4f} secs")

输出:

执行时间:1.0014 秒