"""Format Response Data."""


def prepare_dataload_response(entity_ids, entity_list, key_field_name):
    """Prepare dataload response.

    Prepare dataload response so all the ids will be handled:
    [
        {
            'data': {
                'account_payee_id': 123,
                'account_id': 321,
                ...
            }
        },
        {
            'data': None
        }
    ]
    """
    entity_by_id = {
        entity.get(key_field_name): entity
        for entity
        in entity_list
    }

    result = [
        {
            'data': entity_by_id.get(entity_id, None)
        }
        for entity_id in entity_ids
    ]

    return result