From bd2b65d0688b179776dc8bafed1bde22aef9b8dc Mon Sep 17 00:00:00 2001
From: Hyunjun Kim <kim@hyunjun.kr>
Date: Mon, 11 Mar 2013 13:33:41 +0900
Subject: [PATCH] Improve format_sql(): use sqlparse.format()

---
 flask_debugtoolbar/utils.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/flask_debugtoolbar/utils.py b/flask_debugtoolbar/utils.py
index 3f42c6a..55d4ca2 100644
--- a/flask_debugtoolbar/utils.py
+++ b/flask_debugtoolbar/utils.py
@@ -11,6 +11,12 @@ try:
 except ImportError:
     HAVE_PYGMENTS = False
 
+try:
+    import sqlparse
+    HAVE_SQLPARSE = True
+except ImportError:
+    HAVE_SQLPARSE = False
+
 
 from flask import current_app
 
@@ -46,6 +52,9 @@ def format_fname(value):
     return '<%s>' % path
 
 def format_sql(query, args):
+    if HAVE_SQLPARSE:
+        query = sqlparse.format(query, reindent=True, keyword_case='upper')
+
     if not HAVE_PYGMENTS:
         return query