profiler.py

import cProfile
import pstats


class Profiler:
    def __enter__(self) -> None:
        self.pr = cProfile.Profile()
        self.pr.enable()

    def __exit__(self, exc_type, exc_val, exc_tb):  # type: ignore
        self.pr.disable()
        ps = pstats.Stats(self.pr).sort_stats("cumtime")
        ps.print_stats()