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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cc_py_commons
-Version: 0.5.31
+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.31 → cc_py_commons-0.5.33}/cc_py_commons/config/env.py

@@ -33,6 +33,8 @@ class Config(object):
   BOOKING_ASSISTANT_SNS_TOPIC_ARN = 'arn:aws:sns:us-west-1:791608169866:booking-assistant-flow-test'
   BOOKING_ASSISTANT_SNS_SUBJECT = 'booking-assistant-flow'
   BOOKING_ASSISTANT_SNS_CLASS_NAME = 'com.cargochief.sdk.bookingagent.sns.BookingAssistantFlowNotification'
+  BOOKING_ASSISTANT_QUOTE_CANCEL_SUBJECT = 'quote-cancelled'
+  BOOKING_ASSISTANT_QUOTE_CANCEL_CLASS_NAME = 'com.cargochief.sdk.bookingagent.sns.QuoteCancelNotification'
   IMPORT_MOVEMENTS_SNS_TOPIC_ARN = 'arn:aws:sns:us-west-1:791608169866:mcleod-import-movements-test'
   IMPORT_MOVEMENTS_SNS_SUBJECT = 'import-movements'
   IMPORT_MOVEMENTS_SNS_CLASS_NAME = 'com.cargochief.sdk.c4.sns.mcleod.ImportMovementsNotification'

cc_py_commons-0.5.33/cc_py_commons/sqs/booking_assistant_cancel_quote_notification.py

@@ -0,0 +1,31 @@
+import json
+import traceback
+
+from cc_py_commons.config.env import app_config
+from cc_py_commons.sqs.sqs_service import SqsService
+from cc_py_commons.utils.logger_v2 import logger
+
+class BookingAssistantQuoteCancelNotification:
+
+	def send(self, load_id):
+		'''
+		Sends and event to the Booking Assistant Flow SQS queue specified in the config.
+		Returns the messageId of the enqueued message for later retrieval.
+		'''
+		event = {
+			'loadId': str(load_id),
+			'subject': app_config.BOOKING_ASSISTANT_QUOTE_CANCEL_SUBJECT,
+			'className': app_config.BOOKING_ASSISTANT_QUOTE_CANCEL_CLASS_NAME
+		}
+		logger.debug(f"sending event {event} to {app_config.BOOKING_ASSISTANT_FLOW_QUEUE}")
+
+		try:
+			sqs_service = SqsService()
+			return sqs_service.send(app_config.BOOKING_ASSISTANT_FLOW_QUEUE, json.dumps(event), app_config.BOOKING_ASSISTANT_DELAY_SECONDS)
+		except Exception as e:
+			logger.error(json.dumps({
+				'message': 'Failed to send class BookingAssistantQuoteCancelNotification',
+				'error': str(e),
+				'sta_cktrace': traceback.format_exc(),
+				'event': event
+			}))

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.31 → 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.31
+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.31 → cc_py_commons-0.5.33}/cc_py_commons.egg-info/SOURCES.txt

@@ -107,6 +107,7 @@ cc_py_commons/sns/booking_agent/bid_counter_failure_notification.py
 cc_py_commons/sns/booking_agent/constants.py
 cc_py_commons/sns/booking_agent/send_sns_notification.py
 cc_py_commons/sqs/__init__.py
+cc_py_commons/sqs/booking_assistant_cancel_quote_notification.py
 cc_py_commons/sqs/booking_assistant_flow_event.py
 cc_py_commons/sqs/sqs_service.py
 cc_py_commons/tests/__init__.py

{cc_py_commons-0.5.31 → 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.31',
+    version='0.5.33',
     url='https://github.com/Cargo-Chief/py-commons',
     author='CargoChief',
     author_email='engineering@cargochief.com',

cc_py_commons-0.5.31/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')