def get_pool_data()

in src/pooltool.py [0:0]

• 22 lines of code
• 4 McCabe index (conditional complexity)

def get_pool_data(settings, pool_id, verbose=False):
    session = boto3.session.Session(profile_name=settings['profile'])
    client = session.client('cognito-idp', region_name=settings['region'])
    users = []

    response = client.get_csv_header(
        UserPoolId=pool_id
    )
    csv_headers = response['CSVHeader']

    if verbose:
        print(json.dumps(csv_headers, indent=4))

    response = utilities.list_users_helper(client, pool_id, None)
    users = users + response['Users']

    while 'PaginationToken' in response:
        pagination_token = response['PaginationToken']
        response = utilities.list_users_helper(client, pool_id, pagination_token)
        users = users + response['Users']

    if verbose:
        print(json.dumps(users, indent=4, default=utilities.datetime_converter))

    return {
        "Users": users,
        "CSVHeaders": csv_headers
    }