diff options
Diffstat (limited to 'sqlparse')
| -rw-r--r-- | sqlparse/filters.py | 20 |
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""" |
