Fix parsing of INTO keyword in WHERE clauses (fixes andialbrecht#324).

Andi Albrecht · Andi Albrecht · commit 59033cc0ea75 · 2017-03-02T05:37:20.000+01:00
diff --git a/CHANGELOG b/CHANGELOG
@@ -10,6 +10,7 @@ Bug Fixes
 
 * Fix some edge-cases when parsing invalid SQL statements.
 * Fix indentation of LIMIT (by romainr, issue320).
+* Fix parsing of INTO keyword (issue324).
 
 Internal Changes
 
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
@@ -527,7 +527,7 @@ class Where(TokenList):
     """A WHERE clause."""
     M_OPEN = T.Keyword, 'WHERE'
     M_CLOSE = T.Keyword, ('ORDER', 'GROUP', 'LIMIT', 'UNION', 'EXCEPT',
-                          'HAVING', 'RETURNING')
+                          'HAVING', 'RETURNING', 'INTO')
 
 
 class Case(TokenList):
diff --git a/tests/test_grouping.py b/tests/test_grouping.py
@@ -210,6 +210,14 @@ def test_returning_kw_ends_where_clause():
     assert p.tokens[7].value == 'returning'
 
 
+def test_into_kw_ends_where_clause():  # issue324
+    s = 'select * from foo where a = 1 into baz'
+    p = sqlparse.parse(s)[0]
+    assert isinstance(p.tokens[8], sql.Where)
+    assert p.tokens[9].ttype == T.Keyword
+    assert p.tokens[9].value == 'into'
+
+
 @pytest.mark.parametrize('sql, expected', [
     # note: typecast needs to be 2nd token for this test
     ('select foo::integer from bar', 'integer'),
