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()