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

(cherry picked from commit c0b57bd1c3)
This commit is contained in:
Beto Dealmeida
2024-01-22 11:16:50 -05:00
committed by Michael S. Molina
parent 6cdaf479f2
commit 1d9cfdabd1
17 changed files with 265 additions and 120 deletions

View File

@@ -183,7 +183,7 @@ class Query(
@property
def sql_tables(self) -> list[Table]:
return list(ParsedQuery(self.sql).tables)
return list(ParsedQuery(self.sql, engine=self.db_engine_spec.engine).tables)
@property
def columns(self) -> list["TableColumn"]:
@@ -427,7 +427,9 @@ class SavedQuery(Model, AuditMixinNullable, ExtraJSONMixin, ImportExportMixin):
@property
def sql_tables(self) -> list[Table]:
return list(ParsedQuery(self.sql).tables)
return list(
ParsedQuery(self.sql, engine=self.database.db_engine_spec.engine).tables
)
@property
def last_run_humanized(self) -> str: