From de8f502cbd54fdac1b91042d414dd99502b5c02b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jes=C3=BAs=20Legan=C3=A9s=20Combarro=20=22Piranna=22?=
 <piranna@gmail.com>
Date: Thu, 17 May 2012 23:24:54 +0200
Subject: Added compact filter and tests

---
 sqlparse/filters.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

(limited to 'sqlparse')

diff --git a/sqlparse/filters.py b/sqlparse/filters.py
index 291a613..cc6145a 100644
--- a/sqlparse/filters.py
+++ b/sqlparse/filters.py
@@ -4,11 +4,13 @@ import re
 
 from os.path import abspath, join
 
-from sqlparse import sql
-from sqlparse import tokens as T
-from sqlparse.engine import FilterStack
-from sqlparse.tokens import (Comment, Comparison, Keyword, Name, Punctuation,
-                             String, Whitespace)
+from sqlparse          import sql
+from sqlparse          import tokens as T
+from sqlparse.engine   import FilterStack
+from sqlparse.lexer    import tokenize
+from sqlparse.pipeline import Pipeline
+from sqlparse.tokens   import (Comment, Comparison, Keyword, Name, Punctuation,
+                               String, Whitespace)
 
 
 # --------------------------
@@ -624,3 +626,15 @@ class Limit:
                 return stream[4 - index][1]
 
         return -1
+
+
+def Compact(sql, includePath="sql"):
+    """Function that return a compacted version of the input SQL query"""
+    pipe = Pipeline()
+
+    pipe.append(tokenize)
+    pipe.append(IncludeStatement(includePath))
+    pipe.append(StripComments())
+    pipe.append(StripWhitespace)
+
+    return pipe(sql)
-- 
cgit v1.2.1