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

feat(sqlparse): improve table parsing (#26476)

Browse Source
This commit is contained in:
Beto Dealmeida
2024-01-22 11:16:50 -05:00
committed by GitHub
parent d34874cf2b
commit c0b57bd1c3
17 changed files with 265 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
superset/sql_lab.py
View File

@@ -199,7 +199,7 @@ def execute_sql_statement(
database: Database = query.database
db_engine_spec = database.db_engine_spec
parsed_query = ParsedQuery(sql_statement)
parsed_query = ParsedQuery(sql_statement, engine=db_engine_spec.engine)
if is_feature_enabled("RLS_IN_SQLLAB"):
# There are two ways to insert RLS: either replacing the table with a subquery
# that has the RLS, or appending the RLS to the ``WHERE`` clause. The former is
@@ -219,7 +219,8 @@ def execute_sql_statement(
database.id,
query.schema,
)
)
),
engine=db_engine_spec.engine,
)
sql = parsed_query.stripped()
@@ -409,7 +410,11 @@ def execute_sql_statements(
)
# Breaking down into multiple statements
parsed_query = ParsedQuery(rendered_query, strip_comments=True)
parsed_query = ParsedQuery(
rendered_query,
strip_comments=True,
engine=db_engine_spec.engine,
)
if not db_engine_spec.run_multiple_statements_as_one:
statements = parsed_query.get_statements()
logger.info(