botoresults

boto_results.py

def boto_results(key, fn, *args, **kwargs):
    result = []
    next_token = None
    while True:
        r = fn(*args, **kwargs)
        next_token = r.get("NextToken")
        value = r[key]
        if isinstance(value, (str, bytes)):
            result.append(value)
        elif isinstance(value, Mapping):
            result.append(value)
        else:
            result.extend(value)
        if next_token is None:
            break
        kwargs.update({"NextToken": next_token})
    return result