def handler()

in cloudformation/custom_resources/sqsInvokedFunction/lambda_function.py [0:0]

• 25 lines of code
• 4 McCabe index (conditional complexity)

def handler(event, context):
    for record in event["Records"]:
        payload = json.loads(record["body"])
        product_type = payload["ProductType"]
        product_count = payload["StockCount"]
        bucket, key = payload["S3Uri"].replace("s3://", "").split("/", 1)

        presigned_url = s3.generate_presigned_url(
            "get_object", Params={"Bucket": bucket, "Key": key}, ExpiresIn=3600
        )

        values = {"s3Uri": presigned_url, "count": product_count}

        try:
            mutation = gql_client.execute(
                gql_resource.return_gql(gql_query), variable_values=values
            )
            print(mutation)
        except Exception as e:
            try:
                mutation = gql_client.execute(
                    gql_resource.return_gql(create_bottle_query), variable_values=values
                )
                print(mutation)
            except Exception as ee:
                print(ee)
                raise ee

    return