diff --git a/src/asyncproc.py b/src/asyncproc.py index 623d9c61..428118ab 100644 --- a/src/asyncproc.py +++ b/src/asyncproc.py @@ -148,7 +148,7 @@ def processItems(qUrl, snsTopic, snsRole): limitException = e if(hitLimit): - raise limitException + raise limitException() return totalMessages, jc diff --git a/src/docproc.py b/src/docproc.py index f0d58964..3080511f 100644 --- a/src/docproc.py +++ b/src/docproc.py @@ -6,12 +6,13 @@ def postMessage(client, qUrl, jsonMessage): message = json.dumps(jsonMessage) - client.send_message( + response = client.send_message( QueueUrl=qUrl, MessageBody=message ) print("Submitted message to queue: {}".format(message)) + return response def processRequest(request): @@ -34,7 +35,7 @@ def processRequest(request): qUrl = request['asyncQueueUrl'] if(qUrl): - features = ["Text", "Forms", "Tables"] + features = ["Text", "Forms", "Tables"] jsonMessage = { 'documentId' : documentId, "features" : features, @@ -42,7 +43,8 @@ def processRequest(request): 'objectName' : objectName } client = AwsHelper().getClient('sqs') - postMessage(client, qUrl, jsonMessage) + response = postMessage(client, qUrl, jsonMessage) + return response output = "Completed routing for documentId: {}, object: {}/{}".format(documentId, bucketName, objectName) @@ -76,7 +78,8 @@ def processRecord(record, syncQueueUrl, asyncQueueUrl): request['syncQueueUrl'] = syncQueueUrl request['asyncQueueUrl'] = asyncQueueUrl - processRequest(request) + response = processRequest(request) + return response def lambda_handler(event, context): @@ -94,8 +97,10 @@ def lambda_handler(event, context): if("eventName" in record and record["eventName"] == "INSERT"): if("dynamodb" in record and record["dynamodb"] and "NewImage" in record["dynamodb"]): - processRecord(record, syncQueueUrl, asyncQueueUrl) - + response = processRecord(record, syncQueueUrl, asyncQueueUrl) + return { + 'message': json.dumps(response) + } except Exception as e: print("Faild to process record. Exception: {}".format(e)) diff --git a/src/helper.pyc b/src/helper.pyc new file mode 100644 index 00000000..2318bac3 Binary files /dev/null and b/src/helper.pyc differ