Amazon Pinpoint - Custom Segment Lambda function (Python)

index.py

import random

def handler(event, context):
    # fetch events from input event
    endpoints = event['Endpoints']

    # iterate over endpoints one by one
    for id, endpoint in endpoints.items():
        print("Processing endpoint with id: %s" % id)
        
        # don't include enpoints if not APNS
        if endpoint['ChannelType'] != 'APNS':
            endpoints.pop(id)
            continue
        
        # generate new CreditScore if missing, only for active endpoints
        if endpoint['EndpointStatus'] == 'ACTIVE':
            # add 'Attributes' if missing
            endpoint['Attributes'] = endpoint.get('Attributes', {})
            # add new random CreditScore -> [0, 100]
            endpoint['Attributes']['CreditScore'] = random.randint(0,101)

    print("New endpoints: %s" % endpoints)

    return endpoints