def lambda_handler()

in lambda/processor/handler.py [0:0]

• 27 lines of code
• 6 McCabe index (conditional complexity)

def lambda_handler(event, context):

    print(event)

    records = [
        {
            "body": event_record["body"],
            "timestamp": event_record["attributes"]["SentTimestamp"],
            "message_id": event_record["messageId"],
        }
        for event_record in event["Records"]
        if event_record["body"]
    ]

    prod_endpoint_name = ssm.get_parameter(Name=PROD_ENDPOINT_SSM)["Parameter"]["Value"]
    shadow_endpoint_name = ssm.get_parameter(Name=SHADOW_ENDPOINT_SSM)["Parameter"][
        "Value"
    ]

    if records:
        for record in records:

            body = record["body"][2:].encode().decode()
            prod_probabilty = prediction(prod_endpoint_name, body)

            update_result(prod_endpoint_name, record, prod_probabilty)

            print(prod_probabilty)

            if shadow_endpoint_name != "None":
                shadow_probabilty = prediction(shadow_endpoint_name, body)

                shadow_probabilty = prediction(
                    shadow_endpoint_name, record["body"][2:].encode().decode()
                )

                update_result(shadow_endpoint_name, record, shadow_probabilty)