restore missing uvloop members to uvloop typing

Author: graingert

uvloop/__init__.py

@@ -13,6 +13,9 @@
 __all__ = ('new_event_loop', 'install', 'EventLoopPolicy')
 
 
+_T = _typing.TypeVar("_T")
+
+
 class Loop(__BaseLoop, __asyncio.AbstractEventLoop):  # type: ignore[misc]
     pass
 
@@ -34,10 +37,25 @@ def install() -> None:
     __asyncio.set_event_loop_policy(EventLoopPolicy())
 
 
-def run(main, *, loop_factory=new_event_loop, debug=None, **run_kwargs):
+@_typing.overload
+def run(
+    main: _typing.Coroutine[_typing.Any, _typing.Any, _T],
+    *,
+    debug: _typing.Optional[bool] = ...,
+    loop_factory: _typing.Optional[_typing.Callable[[], Loop]] = ...,
+) -> _T: ...
+
+
+def run(
+    main: _typing.Coroutine[_typing.Any, _typing.Any, _T],
+    *,
+    loop_factory: _typing.Optional[_typing.Callable[[], Loop]] = new_event_loop,
+    debug: _typing.Optional[bool]=None,
+    **run_kwargs: object
+) -> _T:
     """The preferred way of running a coroutine with uvloop."""
 
-    async def wrapper():
+    async def wrapper() -> _T:
         # If `loop_factory` is provided we want it to return
         # either uvloop.Loop or a subtype of it, assuming the user
         # is using `uvloop.run()` intentionally.
@@ -96,7 +114,7 @@ async def wrapper():
         )
 
 
-def _cancel_all_tasks(loop):
+def _cancel_all_tasks(loop: __asyncio.AbstractEventLoop) -> None:
     # Copied from python/cpython
 
     to_cancel = __asyncio.all_tasks(loop)