Tweaked the script

SQSPApiGatewayWebhookListener.py

@@ -3,34 +3,34 @@
 import boto3
 import datetime
 
+TIMESTAMP_FORMAT = '%Y-%m-%dT%H:%M:%S'
 session = boto3.Session()
 s3 = session.resource('s3')
 
-def handler(event, context):
-    '''
-    Parse a proxy API Gateway object's `body` parameter and save to S3.
-    '''
-
-    timestamp = datetime.datetime.now().strftime('%Y-%m-%dT%H:%M:%S')
+'''
+Parse a proxy API Gateway object's `body` parameter and save to S3.
+'''
 
-    if 'body' in event:
-        event_data = event.get('body')
-        data = json.loads(event_data)
-        if 'data' in data:
-            # Test Changes
-            obj = s3.Object(os.environ.get('BUCKET'), f'sqsp/order_{timestamp}.json')
-            res = obj.put(Body=json.dumps(data, indent=4))
-            return {
-                'statusCode': 200,
-                'body': json.dumps({'status': 'success'})
-            }
+def handler(event, context):
+    try:
+        if 'body' in event:
+            event_data = event['body']
+            data = json.loads(event_data)
+            if 'data' in data:
+                timestamp = datetime.datetime.now().strftime(TIMESTAMP_FORMAT)
+                bucket_name = os.environ.get('BUCKET')
+                obj = s3.Object(bucket_name, f'sqsp/order_{timestamp}.json')
+                obj.put(Body=json.dumps(data, indent=4))
+                return {
+                    'statusCode': 200,
+                    'body': json.dumps({'status': 'success'})
+                }
+            else:
+                raise ValueError('Missing data in the request body')
         else:
-            return {
-                'statusCode': 500,
-                'body': json.dumps({'status': 'failed'})
-            }
-    else:
+            raise ValueError('Missing body parameter in the request')
+    except Exception as e:
         return {
             'statusCode': 500,
-            'body': json.dumps({'status': 'failed'})
+            'body': json.dumps({'status': 'failed', 'error': str(e)})
         }