Skip to content

Commit

Permalink
Fixed from BaseSQLQueryRunner to BaseQueryRunner and get_schema inste…
Browse files Browse the repository at this point in the history
…ad of _get_tables
  • Loading branch information
syerushalmy committed Jan 2, 2017
1 parent 163d80a commit 8a3bbca
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions query_runner/cass.py
Original file line number Diff line number Diff line change
@@ -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__)
Expand All @@ -14,7 +14,7 @@
enabled = False


class Cassandra(BaseSQLQueryRunner):
class Cassandra(BaseQueryRunner):
noop_query = "SELECT dateof(now()) FROM system.local"

@classmethod
Expand Down Expand Up @@ -53,13 +53,15 @@ def configuration_schema(cls):
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']
Expand Down

0 comments on commit 8a3bbca

Please sign in to comment.