feat: improve event logging for queries + refactor (#27943)

Co-authored-by: Beto Dealmeida <roberto@dealmeida.net>
This commit is contained in:
Maxime Beauchemin
2024-04-22 16:30:50 -07:00
committed by GitHub
parent 83fedcc9ea
commit cfc440c56c
14 changed files with 169 additions and 132 deletions

View File

@@ -20,7 +20,7 @@ import time
from contextlib import closing
from typing import Any, Optional
from superset import app, security_manager
from superset import app
from superset.models.core import Database
from superset.sql_parse import ParsedQuery
from superset.sql_validators.base import BaseSQLValidator, SQLValidationAnnotation
@@ -54,12 +54,7 @@ class PrestoDBSQLValidator(BaseSQLValidator):
sql = parsed_query.stripped()
# Hook to allow environment-specific mutation (usually comments) to the SQL
if sql_query_mutator := config["SQL_QUERY_MUTATOR"]:
sql = sql_query_mutator(
sql,
security_manager=security_manager,
database=database,
)
sql = database.mutate_sql_based_on_config(sql)
# Transform the final statement to an explain call before sending it on
# to presto to validate