def create_admin_user(self): """ Create admin User. """ prompt(_(u'Create the admin user')) item = list(RandomItemSupplier(1))[0] user_data = UserSerializer(item).data user_data['is_staff'] = True user_data['is_active'] = True user_data['password'] = RandomItemSupplier.ADMIN_PASSWORD info(_(u'Done')) return self.create_user_by_model(**user_data)
def create_clients(self): _drivers_ids = [p.pk for p in self._drivers] _clients = filter(lambda p: p.id not in _drivers_ids, self._persons) how_many = len(_clients) prompt(_(u'Create the clients (%d)' % how_many)) for person in _clients: item = {'person': person.pk, } client_serializer = DriverSerializer(data=item) if client_serializer.is_valid(): client_data = client_serializer.data client = self.create_client_by_api(**client_data) if client: self._clients.append(client) info(_(u'Done'))
def create_drivers(self, how_many): prompt(_(u'Create the drivers (%d)' % how_many)) for person in take_part(self._persons, how_many): license_num_data_list = list(RandomItemSupplier(1)\ .iterator(data_type=RandomItemSupplier.LICENSE_NUM_DATA)) if license_num_data_list: license_num_data = license_num_data_list[0] assert isinstance(license_num_data, dict), \ u'Invalid result type' if 'license_no' in license_num_data: item = {'person': person.pk, 'license_no': license_num_data['license_no']} driver_serializer = DriverSerializer(data=item) if driver_serializer.is_valid(): driver_data = driver_serializer.data driver = self.create_driver_by_api(**driver_data) if driver: self._drivers.append(driver) info(_(u'Done'))
def create_users(self): """ Create ordinary Users. """ prompt(_(u'Create the users (%d)' % self._total_users)) for item in RandomItemSupplier(self._total_users): user_serializer = UserSerializer(data=item) if user_serializer.is_valid(): user_data = user_serializer.data user_data['is_staff'] = False user_data['is_active'] = True user_data['password'] = RandomItemSupplier.USER_PASSWORD user = self.create_user_by_api(**user_data) if user: self._users.append(user) item['user'] = user.pk person_serializer = PersonSerializer(data=item) if person_serializer.is_valid(): person_data = person_serializer.data person = self.create_person_by_api(**person_data) if person: self._persons.append(person) info(_(u'Done'))