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