def handle(self, *args, **options): connection.setup_connection_if_unset() connection.create_keyspace() ConfiguredCassandraUserModelClass = utils.get_cassandra_user_model() self.stdout.write('Sync-ing "{}"'.format(ConfiguredCassandraUserModelClass)) management.sync_table(ConfiguredCassandraUserModelClass) self.stdout.write('Sync-ing "{}"'.format(session_models.CassandraSession)) management.sync_table(session_models.CassandraSession)
def _create_user(self, username=None, auto_first_last_email=False, **kwargs): """ Creates an user instance and inserts it in the database. The received 'kwargs' are passed to YourConfiguredUserModel.create() """ ModelClass = utils.get_cassandra_user_model() if username is None: rnd = uuid.uuid4().hex[0:20] username = "******".format(rnd) if auto_first_last_email: kwargs['first_name'] = kwargs.get('first_name', "John '{}'".format(username)) kwargs['last_name'] = kwargs.get('last_name', "Doe") kwargs['email'] = kwargs.get('email', "{}@example.com".format(username)) cassandra_user = ModelClass.create(username=username, **kwargs) return cassandra_user
def setup(): """ Setup connection, create keyspace and models. Use: call this in the 'setUpClass()' method of the base class of your unittests: class BaseTest(unittest.TestCase, tests_utils.PCassandraTestUtilsMixin): cassandra_connection_done = False @classmethod def setUpClass(cls): unittest.TestCase.setUpClass() if not cls.cassandra_connection_done: tests_utils.PCassandraTestUtilsMixin._full_setup() cls.cassandra_connection_done = True """ setup_connection_and_create_keyspace() ModelClass = utils.get_cassandra_user_model() management.sync_table(ModelClass)
def handle(self, *args, **options): connection.setup_connection_if_unset() ConfiguredCassandraUserModelClass = utils.get_cassandra_user_model() username = input("Enter the username: "******"Password: "******"Password (again): ") if password != password2: print("ERROR: password mismatch") sys.exit(1) user.username = username.strip() user.email = '{}@example.com'.format(username) user.set_password(username) user.is_staff = self.SUPERUSER user.is_superuser = self.SUPERUSER user.save() print("User '{}' created".format(username))
def _get_cassandra_user_model(cls): if not cls._CASSANDRA_USER_MODEL: cls._CASSANDRA_USER_MODEL = utils.get_cassandra_user_model() return cls._CASSANDRA_USER_MODEL