def handler()

in neptune-sagemaker/lambda/add-iam-role-to-neptune.py [0:0]


def handler(event, context):
    logger.info('REQUEST RECEIVED:\n {}'.format(event))
    logger.info('REQUEST RECEIVED:\n {}'.format(context))
    if event['RequestType'] == 'Create':
        logger.info('CREATE')
        
        dbClusterId = event['ResourceProperties']['DBClusterId']
        iamRoleArn = event['ResourceProperties']['NeptuneLoadFromS3IAMRoleArn']
        
        addIamRole(dbClusterId, iamRoleArn)
        sendResponse(event, context, "SUCCESS", { "Message": "Resource creation successful!" })
    elif event['RequestType'] == 'Update':
        logger.info('UPDATE')
        sendResponse(event, context, "SUCCESS", { "Message": "Resource update successful!" })
    elif event['RequestType'] == 'Delete':
        logger.info('DELETE')
        sendResponse(event, context, "SUCCESS", { "Message": "Resource deletion successful!" })
    else:
        logger.info('FAILED!')
        sendResponse(event, context, "FAILED", { "Message": "Unexpected event received from CloudFormation" })