printing sqlalchemy queries

sqlalchemy_query.py

import sqlparse
from sqlalchemy.dialects import postgresql as dialect_postgresql


# ------------------------------------------------------------------------------


def print_query(q):
    """prints a sqlalchemy query"""
    print "-" * 30
    if hasattr(q, 'statement'):
        print "[q.statement.compile %s]" % type(q)
        statement = str(q.statement.compile(dialect=dialect_postgresql.dialect(), compile_kwargs={"literal_binds": True}))
    elif hasattr(q, 'compile'):
        print "[q.compile %s]" % type(q)
        statement = str(q.compile(dialect=dialect_postgresql.dialect(), compile_kwargs={"literal_binds": True}))
    else:
        print "[q %s]" % type(q)
        statement = str(q)
    print sqlparse.format(statement, reindent=True, keyword_case='upper')
    print "-" * 30


# ------------------------------------------------------------------------------