From 1fe5027bd9fd6b0a00667a0968f94346df2d6d6c Mon Sep 17 00:00:00 2001
From: Alex Waygood <Alex.Waygood@Gmail.com>
Date: Mon, 20 Dec 2021 15:35:33 +0000
Subject: [PATCH 1/2] Correct return type of `asynccontextmanager` in 3.10

---
 stdlib/contextlib.pyi | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/stdlib/contextlib.pyi b/stdlib/contextlib.pyi
index df1ca8d28c65..4fe0536f72f9 100644
--- a/stdlib/contextlib.pyi
+++ b/stdlib/contextlib.pyi
@@ -33,14 +33,24 @@ _P = ParamSpec("_P")
 _ExitFunc = Callable[[Optional[Type[BaseException]], Optional[BaseException], Optional[TracebackType]], bool]
 _CM_EF = TypeVar("_CM_EF", AbstractContextManager[Any], _ExitFunc)
 
-class _GeneratorContextManager(AbstractContextManager[_T_co]):
+class ContextDecorator:
     def __call__(self, func: _F) -> _F: ...
 
+class _GeneratorContextManager(AbstractContextManager[_T_co], ContextDecorator): ...
+
 # type ignore to deal with incomplete ParamSpec support in mypy
 def contextmanager(func: Callable[_P, Iterator[_T]]) -> Callable[_P, _GeneratorContextManager[_T]]: ...  # type: ignore[misc]
 
+if sys.version_info >= (3, 10):
+    _AF = TypeVar("_AF", bound=Callable[..., Awaitable[Any]])
+    class AsyncContextDecorator:
+        def __call__(self, func: _AF) -> _AF: ...
+    class _AsyncGeneratorContextManager(AbstractAsyncContextManager[_T_co], AsyncContextDecorator): ...
+elif sys.version_info >= (3, 7):
+    class _AsyncGeneratorContextManager(AbstractAsyncContextManager[_T_co]): ...
+
 if sys.version_info >= (3, 7):
-    def asynccontextmanager(func: Callable[_P, AsyncIterator[_T]]) -> Callable[_P, AbstractAsyncContextManager[_T]]: ...  # type: ignore[misc]
+    def asynccontextmanager(func: Callable[_P, AsyncIterator[_T]]) -> Callable[_P, _AsyncGeneratorContextManager[_T]]: ...  # type: ignore[misc]
 
 class _SupportsClose(Protocol):
     def close(self) -> object: ...
@@ -56,9 +66,6 @@ if sys.version_info >= (3, 10):
     _SupportsAcloseT = TypeVar("_SupportsAcloseT", bound=_SupportsAclose)
     class aclosing(AbstractAsyncContextManager[_SupportsAcloseT]):
         def __init__(self, thing: _SupportsAcloseT) -> None: ...
-    _AF = TypeVar("_AF", bound=Callable[..., Awaitable[Any]])
-    class AsyncContextDecorator:
-        def __call__(self, func: _AF) -> _AF: ...
 
 class suppress(AbstractContextManager[None]):
     def __init__(self, *exceptions: Type[BaseException]) -> None: ...
@@ -72,9 +79,6 @@ class redirect_stdout(AbstractContextManager[_T_io]):
 class redirect_stderr(AbstractContextManager[_T_io]):
     def __init__(self, new_target: _T_io) -> None: ...
 
-class ContextDecorator:
-    def __call__(self, func: _F) -> _F: ...
-
 class ExitStack(AbstractContextManager[ExitStack]):
     def __init__(self) -> None: ...
     def enter_context(self, cm: AbstractContextManager[_T]) -> _T: ...

From c87f169daf48c57b9820db5cea19f53288010f3a Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 20 Dec 2021 15:38:33 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

---
 stdlib/contextlib.pyi | 1 +
 1 file changed, 1 insertion(+)

diff --git a/stdlib/contextlib.pyi b/stdlib/contextlib.pyi
index 4fe0536f72f9..fe72d6dd42d9 100644
--- a/stdlib/contextlib.pyi
+++ b/stdlib/contextlib.pyi
@@ -46,6 +46,7 @@ if sys.version_info >= (3, 10):
     class AsyncContextDecorator:
         def __call__(self, func: _AF) -> _AF: ...
     class _AsyncGeneratorContextManager(AbstractAsyncContextManager[_T_co], AsyncContextDecorator): ...
+
 elif sys.version_info >= (3, 7):
     class _AsyncGeneratorContextManager(AbstractAsyncContextManager[_T_co]): ...