main commit

venv/lib/python3.12/site-packages/starlette/_utils.py

@@ -1,103 +1,74 @@
-from __future__ import annotations
-
+import asyncio
 import functools
 import sys
-from collections.abc import Awaitable, Generator
-from contextlib import AbstractAsyncContextManager, contextmanager
-from typing import Any, Callable, Generic, Protocol, TypeVar, overload
+import typing
+from types import TracebackType
 
-from starlette.types import Scope
-
-if sys.version_info >= (3, 13):  # pragma: no cover
-    from inspect import iscoroutinefunction
-    from typing import TypeIs
+if sys.version_info < (3, 8):  # pragma: no cover
+    from typing_extensions import Protocol
 else:  # pragma: no cover
-    from asyncio import iscoroutinefunction
-
-    from typing_extensions import TypeIs
-
-has_exceptiongroups = True
-if sys.version_info < (3, 11):  # pragma: no cover
-    try:
-        from exceptiongroup import BaseExceptionGroup  # type: ignore[unused-ignore,import-not-found]
-    except ImportError:
-        has_exceptiongroups = False
-
-T = TypeVar("T")
-AwaitableCallable = Callable[..., Awaitable[T]]
+    from typing import Protocol
 
 
-@overload
-def is_async_callable(obj: AwaitableCallable[T]) -> TypeIs[AwaitableCallable[T]]: ...
-
-
-@overload
-def is_async_callable(obj: Any) -> TypeIs[AwaitableCallable[Any]]: ...
-
-
-def is_async_callable(obj: Any) -> Any:
+def is_async_callable(obj: typing.Any) -> bool:
     while isinstance(obj, functools.partial):
         obj = obj.func
 
-    return iscoroutinefunction(obj) or (callable(obj) and iscoroutinefunction(obj.__call__))
+    return asyncio.iscoroutinefunction(obj) or (
+        callable(obj) and asyncio.iscoroutinefunction(obj.__call__)
+    )
 
 
-T_co = TypeVar("T_co", covariant=True)
+T_co = typing.TypeVar("T_co", covariant=True)
 
 
-class AwaitableOrContextManager(Awaitable[T_co], AbstractAsyncContextManager[T_co], Protocol[T_co]): ...
+# TODO: once 3.8 is the minimum supported version (27 Jun 2023)
+# this can just become
+# class AwaitableOrContextManager(
+#     typing.Awaitable[T_co],
+#     typing.AsyncContextManager[T_co],
+#     typing.Protocol[T_co],
+# ):
+#     pass
+class AwaitableOrContextManager(Protocol[T_co]):
+    def __await__(self) -> typing.Generator[typing.Any, None, T_co]:
+        ...  # pragma: no cover
+
+    async def __aenter__(self) -> T_co:
+        ...  # pragma: no cover
+
+    async def __aexit__(
+        self,
+        __exc_type: typing.Optional[typing.Type[BaseException]],
+        __exc_value: typing.Optional[BaseException],
+        __traceback: typing.Optional[TracebackType],
+    ) -> typing.Union[bool, None]:
+        ...  # pragma: no cover
 
 
 class SupportsAsyncClose(Protocol):
-    async def close(self) -> None: ...  # pragma: no cover
+    async def close(self) -> None:
+        ...  # pragma: no cover
 
 
-SupportsAsyncCloseType = TypeVar("SupportsAsyncCloseType", bound=SupportsAsyncClose, covariant=False)
+SupportsAsyncCloseType = typing.TypeVar(
+    "SupportsAsyncCloseType", bound=SupportsAsyncClose, covariant=False
+)
 
 
-class AwaitableOrContextManagerWrapper(Generic[SupportsAsyncCloseType]):
+class AwaitableOrContextManagerWrapper(typing.Generic[SupportsAsyncCloseType]):
     __slots__ = ("aw", "entered")
 
-    def __init__(self, aw: Awaitable[SupportsAsyncCloseType]) -> None:
+    def __init__(self, aw: typing.Awaitable[SupportsAsyncCloseType]) -> None:
         self.aw = aw
 
-    def __await__(self) -> Generator[Any, None, SupportsAsyncCloseType]:
+    def __await__(self) -> typing.Generator[typing.Any, None, SupportsAsyncCloseType]:
         return self.aw.__await__()
 
     async def __aenter__(self) -> SupportsAsyncCloseType:
         self.entered = await self.aw
         return self.entered
 
-    async def __aexit__(self, *args: Any) -> None | bool:
+    async def __aexit__(self, *args: typing.Any) -> typing.Union[None, bool]:
         await self.entered.close()
         return None
-
-
-@contextmanager
-def collapse_excgroups() -> Generator[None, None, None]:
-    try:
-        yield
-    except BaseException as exc:
-        if has_exceptiongroups:  # pragma: no cover
-            while isinstance(exc, BaseExceptionGroup) and len(exc.exceptions) == 1:
-                exc = exc.exceptions[0]
-
-        raise exc
-
-
-def get_route_path(scope: Scope) -> str:
-    path: str = scope["path"]
-    root_path = scope.get("root_path", "")
-    if not root_path:
-        return path
-
-    if not path.startswith(root_path):
-        return path
-
-    if path == root_path:
-        return ""
-
-    if path[len(root_path)] == "/":
-        return path[len(root_path) :]
-
-    return path