def __init__(self, alias, **options):
self.alias = alias
self.hosts = options.get('HOST').split(',')
self.keyspace = options.get('NAME')
self.user = options.get('USER')
self.password = options.get('PASSWORD')
self.options = options.get('OPTIONS', {})
self.cluster_options = self.options.get('connection', {})
self.session_options = self.options.get('session', {})
self.connection_options = {
'lazy_connect': self.cluster_options.pop('lazy_connect', False),
'retry_connect': self.cluster_options.pop('retry_connect', False),
'consistency': self.cluster_options.pop('consistency', None)
}
if self.user and self.password and \
'auth_provider' not in self.cluster_options:
self.cluster_options['auth_provider'] = \
PlainTextAuthProvider(username=self.user,
password=self.password)
self.default = alias == 'default' or \
len(list(get_cassandra_connections())) == 1 or \
self.cluster_options.pop('default', False)
self.register()
def __init__(self, alias, **options):
self.alias = alias
self.hosts = options.get('HOST').split(',')
self.keyspace = options.get('NAME')
self.user = options.get('USER')
self.password = options.get('PASSWORD')
self.options = options.get('OPTIONS', {})
self.cluster_options = self.options.get('connection', {})
self.session_options = self.options.get('session', {})
self.connection_options = {
'lazy_connect': self.cluster_options.pop('lazy_connect', False),
'retry_connect': self.cluster_options.pop('retry_connect', False),
'consistency': self.cluster_options.pop('consistency', None)
}
if self.user and self.password and \
'auth_provider' not in self.cluster_options:
self.cluster_options['auth_provider'] = \
PlainTextAuthProvider(username=self.user,
password=self.password)
self.default = alias == 'default' or \
len(list(get_cassandra_connections())) == 1 or \
self.cluster_options.pop('default', False)
self.register()
def setUp(self):
for alias, _ in get_cassandra_connections():
# Flush the database
call_command('flush', verbosity=0, interactive=False,
database=alias, skip_checks=True,
reset_sequences=False,
allow_cascade=False,
inhibit_post_migrate=True)
def setUp(self):
for alias, _ in get_cassandra_connections():
# Flush the database
call_command('flush',
verbosity=0,
interactive=False,
database=alias,
skip_checks=True,
reset_sequences=False,
allow_cascade=False,
inhibit_post_migrate=True)
def _fixture_teardown(self):
"""
Allow normal django TestCase fixture teardown, but also flush the test
database for each cassandra alias.
"""
super(TestCase, self)._fixture_teardown()
for alias, _ in get_cassandra_connections():
# Flush the database
call_command('flush', verbosity=0, interactive=False,
database=alias, skip_checks=True,
reset_sequences=False,
allow_cascade=False,
inhibit_post_migrate=True)
def _change_cassandra_engine_name(name):
for alias, _ in get_cassandra_connections():
connections[alias].settings_dict['ENGINE'] = name
def ready(self):
from django_cassandra_engine.utils import get_cassandra_connections
for _, conn in get_cassandra_connections():
conn.connect()
def ready(self):
from django_cassandra_engine.utils import get_cassandra_connections
for _, conn in get_cassandra_connections():
conn.connect()
def reset_cassandra_connection():
for _, conn in get_cassandra_connections():
conn.reconnect()
def _databases_names(self, include_mirrors=True):
return [alias for alias, _ in get_cassandra_connections()]
from django_cassandra_engine.utils import get_cassandra_connections
for _, conn in get_cassandra_connections():
conn.connect()
