betodealmeida commented on a change in pull request #11348:
URL:
https://github.com/apache/incubator-superset/pull/11348#discussion_r508954085
##########
File path: superset/sql_parse.py
##########
@@ -111,7 +111,18 @@ def is_select(self) -> bool:
return self._parsed[0].get_type() == "SELECT"
def is_explain(self) -> bool:
- return self.stripped().upper().startswith("EXPLAIN")
+ # Remove comments
+ statements_without_comments = [
+ statement.strip(" \t\n;")
+ for statement in self.stripped().upper().splitlines()
+ if not statement.startswith("--")
+ ]
Review comment:
This will miss some edge cases like:
```sql
-- This is a comment
-- this is another comment but with a space in the front
EXPLAIN SELECT * FROM TABLE
```
You need to `lstrip()` each line as well before testing if it `startswith()`:
```python
statements_without_comments = [
statement.strip(" \t\n;")
for statement in self.stripped().upper().splitlines()
if not statement.lstrip().startswith("--")
]
```
##########
File path: superset/sql_parse.py
##########
@@ -111,7 +111,18 @@ def is_select(self) -> bool:
return self._parsed[0].get_type() == "SELECT"
def is_explain(self) -> bool:
- return self.stripped().upper().startswith("EXPLAIN")
+ # Remove comments
+ statements_without_comments = [
+ statement.strip(" \t\n;")
+ for statement in self.stripped().upper().splitlines()
+ if not statement.startswith("--")
+ ]
+
+ # Explain statements will only be the first statement
+ if statements_without_comments[0].startswith("EXPLAIN"):
+ return True
+
+ return False
Review comment:
You can simplify this:
```python
return statements_without_comments[0].startswith("EXPLAIN")
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
