list dataset records of aws cognito identity pool

list-cognito.py

import boto3

identity_pool_id = "us-east-1:xxxxx"
dataset_name = "dataset-name-xxx"

client_sync = boto3.client("cognito-sync")
client_identity = boto3.client("cognito-identity")


def get_dataset_detail(identity_id):
    global identity_pool_id
    global dataset_name
    global client_sync

    data = client_sync.list_records(
        IdentityPoolId=identity_pool_id,
        DatasetName=dataset_name,
        IdentityId=identity_id,
    )
    if "Records" not in data:
        return None

    res = {}
    for x in data["Records"]:
        if "Key" not in x:
            continue
        res[x["Key"]] = x.get("Value")
    return res


def main():
    global identity_pool_id
    global client_identity

    next_token = None
    while True:
        if next_token:
            data = client_identity.list_identities(
                IdentityPoolId=identity_pool_id, MaxResults=60
            )
        else:
            data = client_identity.list_identities(
                IdentityPoolId=identity_pool_id, MaxResults=60
            )

        if "NextToken" in data:
            next_token = data["NextToken"]

        if not data or "Identities" not in data:
            break

        for x in data["Identities"]:
            identity_id = x["IdentityId"]
            detail = get_dataset_detail(identity_id)
            res = {"identity_id": identity_id, "records": detail}
            print(f"{res}")


if __name__ == "__main__":
    main()