def fetch_all_items()

in aws-cloudknox-config/configrule/CLOUDKNOX_PCI.py [0:0]


def fetch_all_items(client, method, response_key, **kwargs):
    """
    client: The boto3 client
    method: The boto3 method to be fetched (e.g. ecs_client.list_clusters)
    kwargs: The parameter name and value for a given boto3 method (e.g. ecs_client.list_services(cluster=clusterArn))
    """
    if kwargs:
        response = method(**kwargs)
    else:
        response = method()

    items = response[response_key]
    next_token = response.get('nextToken')
    while next_token:
        response = client.method(NextToken=next_token)
        items = items + response[response_key]
        next_token = response.get('nextToken')

    return items