Hotfix support for django_db for celery backend (specially for django_celery_results)

aidin-leo · aidin-leo · commit f4d6e97ce3dd · 2021-01-08T18:55:30.000+03:30
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)()
                 }
             )
 

Original file line number	Diff line number	Diff line change
`@@ -27,11 +27,13 @@ async def receive(self, text_data):`
`27`	`27`	`task_type = text_data_json['type']`
`28`	`28`
`29`	`29`	`if task_type == 'check_task_completion':`
	`30`	`+ from asgiref.sync import sync_to_async`
	`31`	`+ progress = Progress(AsyncResult(self.task_id))`
`30`	`32`	`await self.channel_layer.group_send(`
`31`	`33`	`self.task_id,`
`32`	`34`	`{`
`33`	`35`	`'type': 'update_task_progress',`
`34`		`- 'data': Progress(AsyncResult(self.task_id)).get_info()`
	`36`	`+ 'data': await sync_to_async(progress.get_info, thread_sensitive=True)()`
`35`	`37`	`}`
`36`	`38`	`)`
`37`	`39`