def lambda_handler()

in lambda/start-retraining-with-human-reviewed-data.py [0:0]

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

def lambda_handler(event, context):
    
    # Get the object from the event
    bucketName = event['Records'][0]['s3']['bucket']['name']
    keyName = unquote_plus(event['Records'][0]['s3']['object']['key'])
    fileName = keyName[keyName.rindex('/')+1:keyName.rindex('.')]
    
    print(f"Bucket: {bucketName} key:{keyName} file:{fileName}")

    # Read the S3 Object
    bucket = s3.Bucket(bucketName)
    body = bucket.Object(keyName).get()['Body'].read().decode("utf-8", 'ignore')
    
    # split the body by period to get individual sentences
    body = json.loads(body)
    print(body)
    humanReviewedAnswers = body['humanAnswers'][0]['answerContent']
    feedbackitems = body['inputContent']['comprehendPredictions']
    keys = humanReviewedAnswers.keys()
    print(keys)
    output = ""
    for item, answer in zip(feedbackitems, humanReviewedAnswers):
        utterance = item['utterance']
        prediction = humanReviewedAnswers[answer]
        output = output + prediction + ",\"" + utterance +"\"\n"
    output  = output[:-1]

    kinesis.put_record(DeliveryStreamName=kinesis_delivery_stream,Record={"Data":bytes(output, 'utf-8')})
    
    print(output)

    return 0