utils/queues.py

@@ -19,11 +19,10 @@ class TasksHandlerMixin:
     def poll(self):
         while True:
             response = requests.get(f'{QUEUES_URL}/api/v1/take', headers={'queue': self.queue_name})
-            task = response.get('task')
-            if not task:
+            if response.status_code == 404:
                 time.sleep(0.2)
                 continue
-            data = task.json()
+            data = response.json()
             try:
                 self.process(data['payload'])
             except Exception as exc: