def update_suppliers(data_api_endpoint, data_api_token, updates_path,
updated_by):
client = DataAPIClient(data_api_endpoint, data_api_token)
for update in load_updates(updates_path):
print("Updating {}".format(update))
client.update_supplier(update.pop('id'), update, updated_by)
def suppliers_lot_count(data_api_url, data_api_token):
"""
Generate the CSV
- takes the data api details
- iterates through all suppliers
- foreach supplier hits the draft API to recover the services
- builds CSV row for each supplier
:param data_api_url:
:param data_api_token:
:return:
"""
client = DataAPIClient(data_api_url, data_api_token)
writer = csv.writer(sys.stdout, delimiter=',', quotechar='"')
writer.writerow(headers())
for supplier in client.find_suppliers_iter():
try:
drafts = list()
for draft_service in client.find_draft_services_iter(
supplier['id']):
drafts.append(draft_service)
if drafts:
aggregations = aggregate(drafts)
supplier_row = list()
supplier_row.append(supplier['id'])
supplier_row.append(supplier['name'])
supplier_row.append(supplier.get('dunsNumber', ""))
supplier_row.append(
submitted_count(aggregations.get('iaas', {})))
supplier_row.append(
not_submitted_count(aggregations.get('iaas', {})))
supplier_row.append(
submitted_count(aggregations.get('paas', {})))
supplier_row.append(
not_submitted_count(aggregations.get('paas', {})))
supplier_row.append(
submitted_count(aggregations.get('saas', {})))
supplier_row.append(
not_submitted_count(aggregations.get('saas', {})))
supplier_row.append(
submitted_count(aggregations.get('scs', {})))
supplier_row.append(
not_submitted_count(aggregations.get('scs', {})))
writer.writerow(supplier_row)
except HTTPError as e:
if e.status_code == 404:
# not all suppliers make a declaration so this is fine
pass
else:
raise e
def update_suppliers(data_api_endpoint, data_api_token, users_path):
client = DataAPIClient(data_api_endpoint, data_api_token)
for user in load_users(users_path):
print("Adding {}".format(user))
client.create_user(user)
