diff --git a/superset/sql_parse.py b/superset/sql_parse.py index 2422c1cea..241917a71 100644 --- a/superset/sql_parse.py +++ b/superset/sql_parse.py @@ -34,7 +34,7 @@ class ParsedQuery(object): self._limit = None logging.info('Parsing with sqlparse statement {}'.format(self.sql)) - self._parsed = sqlparse.parse(self.sql) + self._parsed = sqlparse.parse(self.stripped()) for statement in self._parsed: self.__extract_from_token(statement) self._limit = self._extract_limit_from_query(statement) diff --git a/tests/db_engine_specs_test.py b/tests/db_engine_specs_test.py index c04810dae..c2f1713b7 100644 --- a/tests/db_engine_specs_test.py +++ b/tests/db_engine_specs_test.py @@ -196,8 +196,7 @@ class DbEngineSpecsTestCase(SupersetTestCase): FROM table LIMIT 99990""", - """ - SELECT + """SELECT 'LIMIT 777' AS a , b FROM @@ -214,13 +213,12 @@ class DbEngineSpecsTestCase(SupersetTestCase): FROM table LIMIT 99990 ;""", - """ - SELECT + """SELECT 'LIMIT 777' AS a , b FROM table - LIMIT 1000 ;""", + LIMIT 1000""", ) def test_get_datatype(self): @@ -238,8 +236,7 @@ class DbEngineSpecsTestCase(SupersetTestCase): FROM table LIMIT 99990, 999999""", - """ - SELECT + """SELECT 'LIMIT 777' AS a , b FROM @@ -257,8 +254,7 @@ class DbEngineSpecsTestCase(SupersetTestCase): table LIMIT 99990 OFFSET 999999""", - """ - SELECT + """SELECT 'LIMIT 777' AS a , b FROM