diff --git a/superset/sql_parse.py b/superset/sql_parse.py index 30083e484aad7..c8316dd6cc2c7 100644 --- a/superset/sql_parse.py +++ b/superset/sql_parse.py @@ -752,11 +752,21 @@ def _extract_tables_from_sql(self) -> set[Table]: statements = parse(self.stripped(), dialect=self._dialect) except SqlglotError as ex: logger.warning("Unable to parse SQL (%s): %s", self._dialect, self.sql) - dialect = self._dialect or "generic" + + message = ( + "Error parsing near '{highlight}' at line {line}:{col}".format( # pylint: disable=consider-using-f-string + **ex.errors[0] + ) + if isinstance(ex, ParseError) + else str(ex) + ) + raise SupersetSecurityException( SupersetError( error_type=SupersetErrorType.QUERY_SECURITY_ACCESS_ERROR, - message=__(f"Unable to parse SQL ({dialect}): {self.sql}"), + message=__( + f"You may have an error in your SQL statement. {message}" + ), level=ErrorLevel.ERROR, ) ) from ex diff --git a/tests/unit_tests/sql_parse_tests.py b/tests/unit_tests/sql_parse_tests.py index d10c9dbe2507e..593085c4ea568 100644 --- a/tests/unit_tests/sql_parse_tests.py +++ b/tests/unit_tests/sql_parse_tests.py @@ -277,26 +277,30 @@ def test_extract_tables_illdefined() -> None: with pytest.raises(SupersetSecurityException) as excinfo: extract_tables("SELECT * FROM schemaname.") assert ( - str(excinfo.value) == "Unable to parse SQL (generic): SELECT * FROM schemaname." + str(excinfo.value) + == "You may have an error in your SQL statement. Error parsing near '.' at line 1:25" ) with pytest.raises(SupersetSecurityException) as excinfo: extract_tables("SELECT * FROM catalogname.schemaname.") assert ( str(excinfo.value) - == "Unable to parse SQL (generic): SELECT * FROM catalogname.schemaname." + == "You may have an error in your SQL statement. Error parsing near '.' at line 1:37" ) with pytest.raises(SupersetSecurityException) as excinfo: extract_tables("SELECT * FROM catalogname..") assert ( str(excinfo.value) - == "Unable to parse SQL (generic): SELECT * FROM catalogname.." + == "You may have an error in your SQL statement. Error parsing near '.' at line 1:27" ) with pytest.raises(SupersetSecurityException) as excinfo: extract_tables('SELECT * FROM "tbname') - assert str(excinfo.value) == 'Unable to parse SQL (generic): SELECT * FROM "tbname' + assert ( + str(excinfo.value) + == "You may have an error in your SQL statement. Error tokenizing 'SELECT * FROM \"tbnam'" + ) # odd edge case that works assert extract_tables("SELECT * FROM catalogname..tbname") == {