performance_profiling.py

from __future__ import print_function, absolute_import, unicode_literals
from contextlib import contextmanager
from time import time
from io import StringIO
import cProfile
import pstats


@contextmanager
def runtime_logging(task_name, logger=None):
    t_start = time()
    yield
    t_stop = time()

    t_elapsed = t_stop - t_start

    result = 'Took {1:4.3f}s for task: "{0}"'.format(task_name, t_elapsed)

    if logger:
        logger.debug(result)
    else:
        print(result)


@contextmanager
def runtime_profiling(task_name, logger=None):
    pr = cProfile.Profile()
    pr.enable()
    yield
    pr.disable()

    # some in-memory file voodoo
    s = StringIO()
    ps = pstats.Stats(pr, stream=s).sort_stats('tottime')
    ps.print_stats()

    if logger:
        logger.debug(s.getvalue())
    else:
        print(s.getvalue())

usage:

from runtime_profiling import *

with runtime_logging("thing i want to do"):
     # do something you want to profile or call a function here