def lambda_handler()

in functions/analyse_image/app.py [0:0]

• 42 lines of code
• 5 McCabe index (conditional complexity)

def lambda_handler(event, context):
    # Create s3 client
    s3_client = boto3.client('s3')
    #  Create Rekognition Client
    client = boto3.client('rekognition')
    model_arn = os.environ['rekognition_model_project_version_arn']

    for msg in event["Records"]:
        msg_payload = json.loads(msg["body"])
        print("msg_payload: ", msg_payload)
        if "Records" in msg_payload:
            bucket = msg_payload["Records"][0]["s3"]["bucket"]["name"]
            image = msg_payload["Records"][0]["s3"]["object"]["key"].replace("+", " ")
            response = client.detect_custom_labels(
                ProjectVersionArn = model_arn,
                Image={
                    'S3Object': {
                        'Bucket': bucket, 
                        'Name': image}
                    },
                MinConfidence = 70
            )
            # Get the custom labels
            labels = response['CustomLabels']
            # write image to final bucket and delete from incoming bucket
            s3 = boto3.resource('s3')
            finalbucket = os.environ['Final_S3_Bucket_Name']
            copy_source = {
                'Bucket': bucket,
                'Key': image
            }
            random_letters = ''.join(random.choice(string.ascii_letters) for i in range(10))
            put_image_name = random_letters+'-'+image
            s3.meta.client.copy(copy_source, finalbucket, put_image_name)

            # Dump json file with label data in final bucket
            json_object = json.dumps(labels)
            s3_client.put_object(
                Body = str(json_object),
                Bucket = finalbucket,
                Key = put_image_name+'.json'
            )

            # Delete file from incoming s3
            s3_client.delete_object(
                Bucket = bucket,
                Key = image,
            )

        else:
            # Invalid Message - To Be removed from Queue
            print("Invalid msg: ", msg)
            
    return {'status': '200'}