Upstream andialbrecht/sqlparse state of the touched files before f47b549867

This commit is contained in:
upstream committed 2026-09-02 12:14:20 +02:00
1 parent f0a09a4a1b
commit 40155f3cf6
4 files changed
+9 -58

No files matched your search

+6 -20
View File
@@ -188,8 +188,7 @@ def group_period(tlist):
return (pidx, nidx) if valid_next else (pidx, tidx)
_group(tlist, sql.Identifier, match, valid_prev, valid_next, post,
skip_comments=True)
_group(tlist, sql.Identifier, match, valid_prev, valid_next, post)
def group_as(tlist):
@@ -206,8 +205,7 @@ def group_as(tlist):
def post(tlist, pidx, tidx, nidx):
return pidx, nidx
_group(tlist, sql.Identifier, match, valid_prev, valid_next, post,
skip_comments=True)
_group(tlist, sql.Identifier, match, valid_prev, valid_next, post)
def group_assignment(tlist):
@@ -483,16 +481,9 @@ def _group(tlist, cls, match,
post=None,
extend=True,
recurse=True,
depth=0,
skip_comments=False
depth=0
):
"""Groups together tokens that are joined by a middle token. i.e. x < y
When *skip_comments* is ``True`` comments interleaved between the joined
tokens are ignored while looking for the previous and next token, so
e.g. ``a -- c\\n . -- c\\n b`` still groups into a single identifier
(issue712). The comments themselves stay inside the resulting group.
"""
"""Groups together tokens that are joined by a middle token. i.e. x < y"""
if MAX_GROUPING_DEPTH is not None and depth > MAX_GROUPING_DEPTH:
raise SQLParseError(
f"Maximum grouping depth exceeded ({MAX_GROUPING_DEPTH})."
@@ -519,10 +510,10 @@ def _group(tlist, cls, match,
if recurse and token.is_group and not isinstance(token, cls):
_group(token, cls, match, valid_prev, valid_next,
post, extend, True, depth + 1, skip_comments)
post, extend, True, depth + 1)
if match(token):
nidx, next_ = tlist.token_next(tidx, skip_cm=skip_comments)
nidx, next_ = tlist.token_next(tidx)
if prev_ and valid_prev(prev_) and valid_next(next_):
from_idx, to_idx = post(tlist, pidx, tidx, nidx)
grp = tlist.group_tokens(cls, from_idx, to_idx, extend=extend)
@@ -531,9 +522,4 @@ def _group(tlist, cls, match,
pidx, prev_ = from_idx, grp
continue
if skip_comments and imt(token, t=T.Comment, i=sql.Comment):
# Keep prev_ pointing at the real previous token so an interleaved
# comment does not defeat valid_prev (issue712).
continue
pidx, prev_ = tidx, token
+2 -14
View File
@@ -63,20 +63,8 @@ class StripCommentsFilter:
):
# Insert a whitespace to ensure the following SQL produces
# a valid SQL (see #425).
insert_tok = _get_insert_token(token)
# A plain space separator is redundant when a neighbour is
# already whitespace, which happens for comments interleaved
# inside a grouped identifier. A line break carried by the
# comment is always kept.
redundant_space = (
insert_tok.is_whitespace and not insert_tok.is_newline
and ((prev_ is not None and prev_.is_whitespace)
or (next_ is not None and next_.is_whitespace))
)
if (prev_ is not None
and not prev_.match(T.Punctuation, '(')
and not redundant_space):
tlist.tokens.insert(tidx, insert_tok)
if prev_ is not None and not prev_.match(T.Punctuation, '('):
tlist.tokens.insert(tidx, _get_insert_token(token))
tlist.tokens.remove(token)
tidx -= 1
else:
+1 -3
View File
@@ -387,9 +387,7 @@ class TokenList(Token):
A parent object is identified by the first occurring dot.
"""
dot_idx, _ = self.token_next_by(m=(T.Punctuation, '.'))
# skip_cm so a comment between the parent name and the dot (issue712)
# does not get mistaken for the parent name.
_, prev_ = self.token_prev(dot_idx, skip_cm=True)
_, prev_ = self.token_prev(dot_idx)
return remove_quotes(prev_.value) if prev_ is not None else None
def _get_first_name(self, idx=None, reverse=False, keywords=False,
-21
View File
@@ -241,27 +241,6 @@ def test_grouping_identifier_list_with_inline_comments():
assert isinstance(p.tokens[0].tokens[3], sql.Identifier)
def test_grouping_identifier_with_comments():
# issue712: comments between the tokens of a (possibly aliased) dotted
# identifier must not prevent it from being grouped.
s = 'a\n-- c1\n.\n-- c2\nb'
p = sqlparse.parse(s)[0]
assert str(p) == s
assert len([t for t in p.tokens if not t.is_whitespace]) == 1
assert isinstance(p.tokens[0], sql.Identifier)
assert p.tokens[0].get_parent_name() == 'a'
assert p.tokens[0].get_real_name() == 'b'
s = 'ns -- a\n. -- b\n"tbl" -- c\nas -- d\nal'
p = sqlparse.parse(s)[0]
assert str(p) == s
assert len([t for t in p.tokens if not t.is_whitespace]) == 1
assert isinstance(p.tokens[0], sql.Identifier)
assert p.tokens[0].get_parent_name() == 'ns'
assert p.tokens[0].get_real_name() == 'tbl'
assert p.tokens[0].get_alias() == 'al'
def test_grouping_identifiers_with_operators():
p = sqlparse.parse('a+b as c from table where (d-e)%2= 1')[0]
assert len([x for x in p.flatten() if x.ttype == T.Name]) == 5