Fix empty response on sqs dequeue (#14)

ecmwf · Nov 20, 2023 · a833364 · a833364
1 parent 602b3d0
commit a833364
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/polytope_server/common/queue/sqs_queue.py b/polytope_server/common/queue/sqs_queue.py
@@ -1,8 +1,10 @@
 import json
 import logging
-from . import queue
+
 import boto3
+
 from ..metric_collector import SQSQueueMetricCollector
+from . import queue
 
 
 class SQSQueue(queue.Queue):
@@ -30,7 +32,7 @@ def dequeue(self):
             VisibilityTimeout=self.visibility_timeout,  # If processing takes more seconds, message will be read twice
             MaxNumberOfMessages=1,
         )
-        if not response["Messages"]:
+        if "Messages" not in response:
             return None
 
         msg, *remainder = response["Messages"]