Hotfix support for django_db for celery backend (specially for django…

…_celery_results)
czue · Jan 8, 2021 · f4d6e97 · f4d6e97
1 parent a762315
commit f4d6e97
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/celery_progress/websockets/consumers.py b/celery_progress/websockets/consumers.py
@@ -27,11 +27,13 @@ async def receive(self, text_data):
         task_type = text_data_json['type']
 
         if task_type == 'check_task_completion':
+            from asgiref.sync import sync_to_async
+            progress = Progress(AsyncResult(self.task_id))
             await self.channel_layer.group_send(
                 self.task_id,
                 {
                     'type': 'update_task_progress',
-                    'data': Progress(AsyncResult(self.task_id)).get_info()
+                    'data': await sync_to_async(progress.get_info, thread_sensitive=True)()
                 }
             )