QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-07 17:04:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix csv download. (#2036)

Browse Source
This commit is contained in:
Bogdan
2017-01-25 18:06:29 -08:00
committed by GitHub
parent c5c730224e
commit b1bba96d04
4 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
superset/sql_parse.py
View File

@@ -1,6 +1,6 @@
import sqlparse
from sqlparse.sql import IdentifierList, Identifier
from sqlparse.tokens import Keyword, Name
from sqlparse.tokens import DML, Keyword, Name
RESULT_OPERATIONS = {'UNION', 'INTERSECT', 'EXCEPT'}
PRECEDES_TABLE_NAME = {'FROM', 'JOIN', 'DESC', 'DESCRIBE', 'WITH'}
@@ -9,7 +9,6 @@ PRECEDES_TABLE_NAME = {'FROM', 'JOIN', 'DESC', 'DESCRIBE', 'WITH'}
# TODO: some sql_lab logic here.
class SupersetQuery(object):
def __init__(self, sql_statement):
self._tokens = []
self.sql = sql_statement
self._table_names = set()
self._alias_names = set()
@@ -23,9 +22,8 @@ class SupersetQuery(object):
def tables(self):
return self._table_names
# TODO: use sqlparse for this check.
def is_select(self):
return self.sql.upper().startswith('SELECT')
return self._parsed[0].get_type() == 'SELECT'
def stripped(self):
sql = self.sql