QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 23:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump sqlparse to 0.3.0 (#7973)

Browse Source 
* Black

* Bump sqlparse to 0.3.0

* Convert str.format() to f-string
This commit is contained in:
Ville Brofeldt
2019-08-05 17:08:05 +03:00
committed by GitHub
parent b8ca0783ff
commit a1261d7f9a
3 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
superset/sql_parse.py
View File

@@ -195,21 +195,19 @@ class ParsedQuery(object):
if not self._limit:
return f"{self.stripped()}\nLIMIT {new_limit}"
limit_pos = None
tokens = self._parsed[0].tokens
statement = self._parsed[0]
# Add all items to before_str until there is a limit
for pos, item in enumerate(tokens):
for pos, item in enumerate(statement.tokens):
if item.ttype in Keyword and item.value.lower() == "limit":
limit_pos = pos
break
limit = tokens[limit_pos + 2]
_, limit = statement.token_next(idx=limit_pos)
if limit.ttype == sqlparse.tokens.Literal.Number.Integer:
tokens[limit_pos + 2].value = new_limit
limit.value = new_limit
elif limit.is_group:
tokens[limit_pos + 2].value = "{}, {}".format(
next(limit.get_identifiers()), new_limit
)
limit.value = f"{next(limit.get_identifiers())}, {new_limit}"
str_res = ""
for i in tokens:
for i in statement.tokens:
str_res += str(i.value)
return str_res