valitydev/redash
14
0
Fork 0
mirror of https://github.com/valitydev/redash.git synced 2024-11-08 09:53:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed from BaseSQLQueryRunner to BaseQueryRunner and get_schema instead of _get_tables

Browse Source
This commit is contained in:
syerushalmy 2017-01-02 14:47:54 +02:00
parent f54e7b3343
commit ca4db619f2
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
redash/query_runner/cass.py
View File

@ -1,7 +1,7 @@
import json
import logging
from redash.query_runner import BaseSQLQueryRunner, register
from redash.query_runner import BaseQueryRunner, register
from redash.utils import JSONEncoder
logger = logging.getLogger(__name__)
@ -14,7 +14,7 @@ except ImportError:
enabled = False
class Cassandra(BaseSQLQueryRunner):
class Cassandra(BaseQueryRunner):
noop_query = "SELECT dateof(now()) FROM system.local"
@classmethod
@ -53,13 +53,15 @@ class Cassandra(BaseSQLQueryRunner):
def type(cls):
return "Cassandra"
def _get_tables(self, schema):
def get_schema(self, get_stats=False):
query = """
SELECT columnfamily_name, column_name FROM system.schema_columns where keyspace_name ='{}';
""".format(self.configuration['keyspace'])
results, error = self.run_query(query, None)
results = json.loads(results)
schema = {}
for row in results['rows']:
table_name = row['columnfamily_name']
column_name = row['column_name']