Flask SQLAlchemy - print stack trace for every query

sqlalchemy_trace.py

import sqlalchemy.event

def show_query_stack_traces():
    base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))

    def catch_queries_callback(connection, cursor, query, *_):
        print query
        stacktrace = [
            (file_path, val1, val2, line_contents)
            for (file_path, val1, val2, line_contents) in traceback.extract_stack()
            if file_path.startswith(base_path)
        ]

        for i in stacktrace:
            print "\t", i
        print "\n"
    
    sqlalchemy.event.listen(db.engine, "before_cursor_execute", catch_queries_callback)