profile decorator cProfile - Python 3.9

profile.py

import cProfile, pstats, io

def profile(fun):
    """
    print file-function call, number of calls, cumtime, percall(second percall)
    How to use:

    @profile
    def function_to_analize():
        ...

    if __name__ == "__main__":
        function_to_analize()
    """
    def inner(*args, **kargs):
        pr = cProfile.Profile()
        pr.enable()
        retval = fun(*args, **kargs)
        pr.disable()
        s = io.StringIO()
        sortby = 'cumulative'
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        print(s.getvalue())
        return retval

    return inner