Decorator to add profiling to Python code using CProfile

profiler.py

import cProfile
import io
import pstats

is_profiler_enabled = False


def profile(func):
    def enabled_wrapper(*args, **kwargs):
        pr = cProfile.Profile()
        pr.enable()
        retval = func(*args, **kwargs)
        pr.disable()
        s = io.StringIO()
        sortby = pstats.SortKey.CUMULATIVE  # 'cumulative'
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        print(s.getvalue())
        return retval

    def disabled_wrapper(*args, **kwargs):
        retval = func(*args, **kwargs)
        return retval

    return enabled_wrapper if is_profiler_enabled else disabled_wrapper