Simple context manager for python cProfiler

context_manager_profiler.py

import cProfile, pstats, io


class MyProfiler:
    def __init__(self):
        self.pr = cProfile.Profile()
        self.pr.enable()

    def __enter__(self):
        return self.pr

    def __exit__(self, *args):
        self.pr.disable()
        s = io.StringIO()
        ps = pstats.Stats(self.pr, stream=s).sort_stats('cumulative')
        ps.print_stats()
        print(s.getvalue())