A decorator to run profiler in Python

profiler_decorator.py

"""
with Python 3.8, you can use context manager with cProfile.Profile()
"""

def profile(func, filename="stats"):
    def wrapper(*args, **kwargs):
        import cProfile, pstats, io
        from pstats import SortKey
        pr = cProfile.Profile()
        pr.enable()
        v = func(*args, **kwargs)
        pr.disable()
        s = io.StringIO()
        sortby = SortKey.CUMULATIVE
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        with open(filename, 'a') as f:
            f.write(s.getvalue())
        return v
    return wrapper