pretty print your SQL

prettySQL.py

#!/usr/bin/env python
import argparse
import sys
import sqlparse

from pygments import highlight
from pygments.lexers import get_lexer_by_name
from pygments.formatters import TerminalTrueColorFormatter

def prettyPrint(raw_sql):
    lexer = get_lexer_by_name('sql', stripall=True)
    formatter = TerminalTrueColorFormatter()
    formatted_sql = sqlparse.format(raw_sql, encoding=None, keyword_case='upper', identifier_case='lower', reindent=True)

    return highlight(formatted_sql, lexer, formatter)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Pretty print sql.")
    parser.add_argument('sql', nargs='?', help='sql as a string')
    parser.add_argument('-f', '--file', help='read from file.')

    args = parser.parse_args()

    if args.file:
        sql = open(args.file, 'r')
    elif args.sql:
        sql = args.sql
    else:
        sql = sys.stdin

    print(prettyPrint(sql))

I do: pbpaste | sqlpp a lot. I wonder if there's an easy way to support that.

I will see what I can do.