diff options
Diffstat (limited to 'sqlparse/engine/grouping.py')
| -rw-r--r-- | sqlparse/engine/grouping.py | 13 |
1 files changed, 13 insertions, 0 deletions
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 |