profile_cm.py

from pathlib import Path
import cProfile


class Profile:

    def __init__(self, path: Path):
        self.profiler = None
        self.path = path

    def __enter__(self):
        self.profiler = cProfile.Profile()
        self.profiler.enable()
        return self

    def __exit__(self, *args, **kwargs):
        self.profiler.dump_stats(self.path)
        self.profiler = None