Context manager to profile sqlalchemy query performance

sqlalchemy_query_profiling.py

import cProfile
import StringIO
import pstats
import contextlib


@contextlib.contextmanager
def profiled():
    pr = cProfile.Profile()
    pr.enable()
    yield
    pr.disable()
    s = StringIO.StringIO()
    ps = pstats.Stats(pr, stream=s).sort_stats('cumulative')
    ps.print_stats()
    # uncomment this to see who's calling what
    # ps.print_callers()
    print s.getvalue()


# To use, wrap the execution of a sqlalchemy query object, like so:
with profiled():
    data = session.query(TableModel).all()