Quick functions to easily profile Python code

profile.py

import monotonic
from time import clock as cpu_time
import cProfile, pstats
try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO

def time_it(label):
    start_total = monotonic.monotonic()
    start_cpu = cpu_time()

    def split(name=None):
        diff_total = monotonic.monotonic() - start_total
        diff_cpu = cpu_time() - start_cpu
        desc = '%s (%s)' % (label, name) if name else label
        print('%s: %f (%f)' % (desc, diff_total, diff_cpu))

    return split

def profile():
    pr = cProfile.Profile()
    pr.enable()

    def stop_profile():
        pr.disable()
        s = StringIO()
        sortby = 'cumulative'
        ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
        ps.print_stats()
        print(s.getvalue())

    return stop_profile