python-codex 0.2.7__py3-none-any.whl → 0.3.0__py3-none-any.whl

pycodex/compat.py

@@ -11,6 +11,7 @@ except ImportError:  # pragma: no cover - Python 3.6 path
     class ThreadingHTTPServer(ThreadingMixIn, HTTPServer):
         daemon_threads = True
 
+
 try:
     from importlib import metadata as importlib_metadata
 except ImportError:  # pragma: no cover - Python 3.6 path
@@ -20,6 +21,7 @@ try:
     from typing import Literal, Protocol, TypeAlias
 except ImportError:  # pragma: no cover - Python 3.6 path
     from typing_extensions import Literal, Protocol  # type: ignore
+
     try:
         from typing_extensions import TypeAlias  # type: ignore
     except ImportError:  # pragma: no cover - old typing_extensions
@@ -34,6 +36,20 @@ def _get_running_loop_compat():
 
 
 def patch_asyncio():
+    if not hasattr(asyncio, "current_task"):
+        asyncio.current_task = asyncio.Task.current_task
+
+    if not hasattr(asyncio, "all_tasks"):
+
+        def all_tasks(loop=None):
+            if loop is None:
+                loop = asyncio.get_running_loop()
+            return {
+                task for task in asyncio.Task.all_tasks(loop=loop) if not task.done()
+            }
+
+        asyncio.all_tasks = all_tasks
+
     if not hasattr(asyncio, "create_task"):
         asyncio.create_task = asyncio.ensure_future
 
@@ -41,6 +57,7 @@ def patch_asyncio():
         asyncio.get_running_loop = _get_running_loop_compat
 
     if not hasattr(asyncio, "to_thread"):
+
         async def to_thread(func, *args, **kwargs):
             loop = asyncio.get_event_loop()
             call = functools.partial(func, *args, **kwargs)
@@ -49,17 +66,14 @@ def patch_asyncio():
         asyncio.to_thread = to_thread
 
     if not hasattr(asyncio, "run"):
+
         def run(main):
             loop = asyncio.new_event_loop()
             try:
                 asyncio.set_event_loop(loop)
                 return loop.run_until_complete(main)
             finally:
-                all_tasks = getattr(asyncio.Task, "all_tasks", None)
-                if all_tasks is not None:
-                    pending = all_tasks(loop=loop)
-                else:
-                    pending = asyncio.all_tasks(loop)
+                pending = asyncio.all_tasks(loop)
                 for task in pending:
                     task.cancel()
                 if pending: