summaryrefslogtreecommitdiff
path: root/sqlparse
diff options
context:
space:
mode:
Diffstat (limited to 'sqlparse')
-rw-r--r--sqlparse/filters.py20
1 files changed, 20 insertions, 0 deletions
diff --git a/sqlparse/filters.py b/sqlparse/filters.py
index 23d06e1..d54105e 100644
--- a/sqlparse/filters.py
+++ b/sqlparse/filters.py
@@ -75,6 +75,26 @@ class StripComments(Filter):
yield token_type, value
+def StripWhitespace(stream):
+ """Strip the whitespaces from a stream"""
+ last_type = None
+
+ for token_type, value in stream:
+ if last_type == None:
+ if token_type not in Whitespace + Punctuation:
+ yield token_type, value
+ last_type = token_type
+
+ else:
+ if token_type in Whitespace:
+ if last_type not in Whitespace:
+ yield token_type, ' '
+ else:
+ yield token_type, value
+
+ last_type = token_type
+
+
class IncludeStatement(Filter):
"""Filter that enable a INCLUDE statement"""