From fa7def01d00fc7c5c3047b077bc0f1a4cdc1d10a Mon Sep 17 00:00:00 2001 From: Fredy Wijaya Date: Thu, 13 Dec 2018 16:44:29 -0600 Subject: Fix formatting on INSERT (fixes #329) This patch fixes the formatting on INSERT by creating a new instance of sql.Values to group all the values. SQL: insert into foo values (1, 'foo'), (2, 'bar'), (3, 'baz') Before: insert into foo values (1, 'foo'), (2, 'bar'), (3, 'baz') After: insert into foo values (1, 'foo'), (2, 'bar'), (3, 'baz') --- sqlparse/engine/grouping.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'sqlparse/engine') diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py index 5ff819c..afc9123 100644 --- a/sqlparse/engine/grouping.py +++ b/sqlparse/engine/grouping.py @@ -327,6 +327,18 @@ def align_comments(tlist): tidx, token = tlist.token_next_by(i=sql.Comment, idx=tidx) +def group_values(tlist): + tidx, token = tlist.token_next_by(m=(T.Keyword, 'VALUES')) + start_idx = tidx + end_idx = -1 + while token: + if isinstance(token, sql.Parenthesis): + end_idx = tidx + tidx, token = tlist.token_next(tidx) + if end_idx != -1: + tlist.group_tokens(sql.Values, start_idx, end_idx, extend=True) + + def group(stmt): for func in [ group_comments, @@ -354,6 +366,7 @@ def group(stmt): align_comments, group_identifier_list, + group_values, ]: func(stmt) return stmt -- cgit v1.2.1