Measure code execution time within a block using a Python context manager

catchtime_context_manager.py

from time import perf_counter


class catchtime:
    def __enter__(self):
        self._time_start = perf_counter()
        return self

    def __exit__(self, type, value, traceback):
        self.elapsed = perf_counter() - self._time_start

Usage example:

with catchtime() as t:
    time.sleep(0.42)

print(t.elapsed)
# 0.4206...