feat: improve _extract_tables_from_sql (#26748)

This commit is contained in:
Beto Dealmeida
2024-03-18 13:02:58 -04:00
committed by Michael S. Molina
parent d4314a92d8
commit 94f677850c
4 changed files with 53 additions and 8 deletions

View File

@@ -25,6 +25,7 @@ from dataclasses import dataclass
from typing import Any, cast, Optional
import sqlparse
from flask_babel import gettext as __
from sqlalchemy import and_
from sqlglot import exp, parse, parse_one
from sqlglot.dialects import Dialects
@@ -55,7 +56,11 @@ from sqlparse.tokens import (
)
from sqlparse.utils import imt
from superset.exceptions import QueryClauseValidationException
from superset.errors import ErrorLevel, SupersetError, SupersetErrorType
from superset.exceptions import (
QueryClauseValidationException,
SupersetSecurityException,
)
from superset.utils.backports import StrEnum
try:
@@ -287,9 +292,16 @@ class ParsedQuery:
"""
try:
statements = parse(self.stripped(), dialect=self._dialect)
except SqlglotError:
except SqlglotError as ex:
logger.warning("Unable to parse SQL (%s): %s", self._dialect, self.sql)
return set()
dialect = self._dialect or "generic"
raise SupersetSecurityException(
SupersetError(
error_type=SupersetErrorType.QUERY_SECURITY_ACCESS_ERROR,
message=__(f"Unable to parse SQL ({dialect}): {self.sql}"),
level=ErrorLevel.ERROR,
)
) from ex
return {
table