jprdonnelly/redash
1
0
Fork 0
mirror of https://github.com/getredash/redash.git synced 2025-12-25 01:03:20 -05:00
Code Issues Projects Releases Wiki Activity

Fix code style

Browse Source
This commit is contained in:
Maxime Fouilleul
2017-04-19 22:14:27 +02:00
parent 519fb49f6a
commit 5d7795ca47
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
redash/query_runner/cass.py
View File

@@ -14,12 +14,14 @@ try:
except ImportError:
enabled = False
class CassandraJSONEncoder(JSONEncoder):
def default(self, o):
if isinstance(o, uuid.UUID):
return str(o)
return super(CassandraJSONEncoder, self).default(o)
class Cassandra(BaseQueryRunner):
noop_query = "SELECT dateof(now()) FROM system.local"
@@ -92,9 +94,14 @@ class Cassandra(BaseQueryRunner):
if self.configuration.get('username', '') and self.configuration.get('password', ''):
auth_provider = PlainTextAuthProvider(username='{}'.format(self.configuration.get('username', '')),
password='{}'.format(self.configuration.get('password', '')))
connection = Cluster([self.configuration.get('host', '')], auth_provider=auth_provider, protocol_version=self.configuration.get('protocol', ''), cql_version=self.configuration.get('cqlversion', ''))
connection = Cluster([self.configuration.get('host', '')],
auth_provider=auth_provider,
protocol_version=self.configuration.get('protocol', ''),
cql_version=self.configuration.get('cqlversion', ''))
else:
connection = Cluster([self.configuration.get('host', '')], protocol_version=self.configuration.get('protocol', ''), cql_version=self.configuration.get('cqlversion', ''))
connection = Cluster([self.configuration.get('host', '')],
protocol_version=self.configuration.get('protocol', ''),
cql_version=self.configuration.get('cqlversion', ''))
session = connection.connect()
session.set_keyspace(self.configuration['keyspace'])
logger.debug("Cassandra running query: %s", query)