def lambda_handler()

in deploy_code/multipagepdfa2i_kickoff/lambda_function.py [0:0]

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

def lambda_handler(event, context):
    try:
        # this is the sqs payload
        for record in event["Records"]:
            # these are the s3 payload
            for cur_record in json.loads(record["body"])["Records"]:
                data = extract_event_data(cur_record)
                extension = data["key"][-3:].lower()
                if extension == "pdf" or extension == "png" or extension == "jpg":
                    payload = {
                        "id": data["id"],
                        "bucket": data["bucket"],
                        "key": data["key"],
                        "extension": extension
                    }
                    response = start_step_function(payload)
            # delete message from sqs
            client = boto3.client('sqs')
            response = client.delete_message(
                QueueUrl=os.environ['sqs_url'],
                ReceiptHandle=record["receiptHandle"]
            )
    except:
        return "something didn't go right...."