Merge pull request #42 from meedan/cv2-3435-add-more-prefix

CV2-3435 add another prefix call
meedan · Oct 11, 2023 · ab5621e · ab5621e
2 parents ff8610e + 3cbff5e
commit ab5621e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/queue/queue.py b/lib/queue/queue.py
@@ -5,7 +5,7 @@
 import boto3
 import botocore
 
-from lib.helpers import get_environment_setting
+from lib.helpers import get_setting, get_environment_setting
 from lib.logger import logger
 from lib import schemas
 SQS_MAX_BATCH_SIZE = 10
@@ -154,5 +154,6 @@ def push_message(self, queue_name: str, message: schemas.Message) -> schemas.Mes
         """
         Actual SQS logic for pushing a message to a queue
         """
-        self.find_queue_by_name(queue_name).send_message(MessageBody=json.dumps(message.dict()))
+        queue_prefix = get_setting("", "QUEUE_PREFIX").replace(".", "__")
+        self.find_queue_by_name(f"{queue_prefix}{queue_name}").send_message(MessageBody=json.dumps(message.dict()))
         return message