fix: get QueueArn attribute in a way that works

Doist · Apr 25, 2024 · 8d968b4 · 8d968b4
1 parent 2f09c7d
commit 8d968b4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/sqs_workers/core.py b/sqs_workers/core.py
@@ -49,14 +49,16 @@ def __init__(self, sqs_env, dead_letter_queue_name, max_receive_count):
         self.max_receive_count = max_receive_count
 
     def __json__(self) -> str:
-        queue = self.sqs_env.sqs_resource.get_queue_by_name(
-            QueueName=self.sqs_env.get_sqs_queue_name(self.dead_letter_queue_name)
+        queue_name = self.sqs_env.get_sqs_queue_name(self.dead_letter_queue_name)
+        queue = self.sqs_env.sqs_resource.get_queue_by_name(QueueName=queue_name)
+        queue_attrs = self.sqs_env.sqs_client.get_queue_attributes(
+            QueueUrl=queue.url,
+            AttributeNames=["QueueArn"],
         )
-        target_arn = queue.attributes["QueueArn"]
         # Yes, it's double-encoded JSON :-/
         return json.dumps(
             {
-                "deadLetterTargetArn": target_arn,
+                "deadLetterTargetArn": queue_attrs["Attributes"]["QueueArn"],
                 "maxReceiveCount": str(self.max_receive_count),
             }
         )