cc-py-commons 0.5.32__tar.gz → 0.5.33__tar.gz

{cc_py_commons-0.5.32 → cc_py_commons-0.5.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cc_py_commons
-Version: 0.5.32
+Version: 0.5.33
 Summary: Common code for microservices
 Home-page: https://github.com/Cargo-Chief/py-commons
 Author: CargoChief

cc_py_commons-0.5.33/cc_py_commons/sqs/sqs_service.py

@@ -0,0 +1,86 @@
+import json
+from uuid import uuid4
+import boto3
+
+class SqsService:
+	def send(self, queue_url, message, delaySeconds=0, messageGroupId=None):
+		'''
+		Sends a message to the specified queue and returns the messageId. 
+		The messageId can be used to lookup and delete the message in the queue.
+		'''
+		sqs = boto3.client('sqs')
+		if messageGroupId:
+			response = sqs.send_message(
+				QueueUrl=queue_url,
+				MessageBody=message,
+				DelaySeconds=delaySeconds,
+				MessageAttributes={},
+				MessageGroupId=messageGroupId
+			)
+		else:
+			response = sqs.send_message(
+				QueueUrl=queue_url,
+				MessageBody=message,
+				DelaySeconds=delaySeconds,
+				MessageAttributes={}
+			)
+		return response.get('MessageId')
+
+def send_batches(queue_url, messages):
+	'''
+	Sends all messages to the specified queue in groups of 10 (the limit of send_batch)
+	The messageId can be used to lookup and delete the message in the queue.
+	Args:
+		queue_url (str): The URL of the Amazon SQS queue
+		messages (list): List of messages to be sent
+
+	Returns:
+		dict: Results containing 'successful' and 'failed' message lists
+	'''
+	results = {
+		'successful': [],
+		'failed': []
+	}
+
+	if not messages:
+		return results
+
+	sqs = boto3.client('sqs')
+	entries = [
+		{
+			# each message must have a unique id
+			'Id': str(uuid4()),
+			'MessageBody': json.dumps(message),
+		}
+		for message in messages
+	]
+
+	# SQS can only process 10 messages per batch
+	batch_size = 10
+
+	# Split messages into batches of 10
+	for i in range(0, len(entries), batch_size):
+		batch = entries[i:i + batch_size]
+
+		try:
+			response = sqs.send_message_batch(
+				QueueUrl=queue_url,
+				Entries=batch
+			)
+
+			# Track successful and failed messages
+			if 'Successful' in response:
+				results['successful'].extend(response['Successful'])
+			if 'Failed' in response:
+				results['failed'].extend(response['Failed'])
+
+		except Exception as e:
+			# Mark all messages in failed batch
+			failed_batch = [{
+				'Id': entry['Id'],
+				'Error': str(e),
+				'Message': entry['MessageBody']
+			} for entry in batch]
+			results['failed'].extend(failed_batch)
+
+	return results

{cc_py_commons-0.5.32 → cc_py_commons-0.5.33}/cc_py_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cc-py-commons
-Version: 0.5.32
+Version: 0.5.33
 Summary: Common code for microservices
 Home-page: https://github.com/Cargo-Chief/py-commons
 Author: CargoChief

{cc_py_commons-0.5.32 → cc_py_commons-0.5.33}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 setup(
     name='cc_py_commons',
     description='Common code for microservices',
-    version='0.5.32',
+    version='0.5.33',
     url='https://github.com/Cargo-Chief/py-commons',
     author='CargoChief',
     author_email='engineering@cargochief.com',

cc_py_commons-0.5.32/cc_py_commons/sqs/sqs_service.py

@@ -1,25 +0,0 @@
-import boto3
-
-class SqsService:
-	def send(self, queue_url, message, delaySeconds=0, messageGroupId=None):
-		'''
-		Sends a message to the specified queue and returns the messageId. 
-		The messageId can be used to lookup and delete the message in the queue.
-		'''
-		sqs = boto3.client('sqs')
-		if messageGroupId:
-			response = sqs.send_message(
-				QueueUrl=queue_url,
-				MessageBody=message,
-				DelaySeconds=delaySeconds,
-				MessageAttributes={},
-				MessageGroupId=messageGroupId
-			)
-		else:
-			response = sqs.send_message(
-				QueueUrl=queue_url,
-				MessageBody=message,
-				DelaySeconds=delaySeconds,
-				MessageAttributes={}
-			)
-		return response.get('MessageId')