From 47706b9d02b8a444e5b712d99c9b511ce91a7992 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jes=C3=BAs=20Legan=C3=A9s=20Combarro=20=22Piranna=22?= Date: Tue, 14 Feb 2012 12:47:54 +0100 Subject: Added StripWhitespace filter --- sqlparse/filters.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'sqlparse') 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""" -- cgit v1.2.1