Strip leading and trailing whitespaces from splitted statements.

andialbrecht · andialbrecht · commit abbcdcf9e135 · 2013-05-15T08:14:12.000+02:00
diff --git a/CHANGES b/CHANGES
@@ -6,6 +6,8 @@ Enhancements
 * Add STRAIGHT_JOIN statement (by Yago Riveiro).
 * Function.get_parameters() now returns the parameter if only one parameter is
   given (issue94, by wayne.wuw).
+* sqlparse.split() now removes leading and trailing whitespaces from splitted
+  statements.
 
 
 Release 0.1.7 (Apr 06, 2013)
diff --git a/sqlparse/__init__.py b/sqlparse/__init__.py
@@ -67,7 +67,7 @@ def split(sql, encoding=None):
     """
     stack = engine.FilterStack()
     stack.split_statements = True
-    return [unicode(stmt) for stmt in stack.run(sql, encoding)]
+    return [unicode(stmt).strip() for stmt in stack.run(sql, encoding)]
 
 
 from sqlparse.engine.filter import StatementFilter
diff --git a/tests/test_split.py b/tests/test_split.py
@@ -131,3 +131,10 @@ def test_encoding_parsestream(self):
         stream = StringIO("SELECT 1; SELECT 2;")
         stmts = list(sqlparse.parsestream(stream))
         self.assertEqual(type(stmts[0].tokens[0].value), unicode)
+
+
+def test_split_simple():
+    stmts = sqlparse.split('select * from foo; select * from bar;')
+    assert len(stmts) == 2
+    assert stmts[0] == 'select * from foo;'
+    assert stmts[1] == 'select * from bar;'
