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