Revert removing of SyncToAsync.get_current_task

django · Mar 21, 2024 · 6ba6881 · 6ba6881
1 parent 852344e
commit 6ba6881
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/asgiref/sync.py b/asgiref/sync.py
@@ -519,6 +519,16 @@ def thread_handler(self, loop, exc_info, task_context, func, *args, **kwargs):
         else:
             return func(*args, **kwargs)
 
+    @staticmethod
+    def get_current_task() -> Optional["asyncio.Task[Any]"]:
+        """
+        Implementation of asyncio.current_task()
+        that returns None if there is no task.
+        """
+        try:
+            return asyncio.current_task()
+        except RuntimeError:
+            return None
 
 @overload
 def async_to_sync(