QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-10 02:15:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: extract tables from jinja (#34307)

Browse Source
This commit is contained in:
JUST.in DO IT
2025-07-25 05:06:15 -07:00
committed by GitHub
parent fb035f4210
commit 5eb61ec1b6
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
superset/sql_parse.py
View File

@@ -943,11 +943,11 @@ def extract_tables_from_jinja_sql(sql: str, database: Database) -> set[Table]:
)
processor = get_template_processor(database)
template = processor.env.parse(sql)
ast = processor.env.parse(sql)
tables = set()
for node in template.find_all(nodes.Call):
for node in ast.find_all(nodes.Call):
if isinstance(node.node, nodes.Getattr) and node.node.attr in (
"latest_partition",
"latest_sub_partition",
@@ -972,7 +972,9 @@ def extract_tables_from_jinja_sql(sql: str, database: Database) -> set[Table]:
node.data = "NULL"
# re-render template back into a string
rendered_sql = Template(template).render(processor.get_context())
code = processor.env.compile(ast)
template = Template.from_code(processor.env, code, globals=processor.env.globals)
rendered_sql = template.render(processor.get_context())
return (
tables