strawberry-graphql 0.237.3__py3-none-any.whl → 0.238.0__py3-none-any.whl

strawberry/aiohttp/views.py

@@ -15,6 +15,7 @@ from typing import (
 )
 
 from aiohttp import web
+from aiohttp.multipart import BodyPartReader
 from strawberry.aiohttp.handlers import (
     GraphQLTransportWSHandler,
     GraphQLWSHandler,
@@ -60,6 +61,7 @@ class AioHTTPRequestAdapter(AsyncHTTPRequestAdapter):
         files: Dict[str, Any] = {}
 
         async for field in reader:
+            assert isinstance(field, BodyPartReader)
             assert field.name
 
             if field.filename:

strawberry/extensions/tracing/sentry.py

@@ -24,7 +24,7 @@ class SentryTracingExtension(SchemaExtension):
         execution_context: Optional[ExecutionContext] = None,
     ) -> None:
         warnings.warn(
-            "The Sentry tracing extension is deprecated, please update to sentry>=1.32.0",
+            "The Sentry tracing extension is deprecated, please update to sentry-sdk>=1.32.0",
             DeprecationWarning,
             stacklevel=2,
         )

{strawberry_graphql-0.237.3.dist-info → strawberry_graphql-0.238.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strawberry-graphql
-Version: 0.237.3
+Version: 0.238.0
 Summary: A library for creating GraphQL APIs
 Home-page: https://strawberry.rocks/
 License: MIT
@@ -35,7 +35,6 @@ Provides-Extra: pydantic
 Provides-Extra: pyinstrument
 Provides-Extra: quart
 Provides-Extra: sanic
-Provides-Extra: starlite
 Requires-Dist: Django (>=3.2) ; extra == "django"
 Requires-Dist: aiohttp (>=3.7.4.post0,<4.0.0) ; extra == "aiohttp"
 Requires-Dist: asgiref (>=3.2,<4.0) ; extra == "django" or extra == "channels"
@@ -59,7 +58,6 @@ Requires-Dist: quart (>=0.19.3) ; extra == "quart"
 Requires-Dist: rich (>=12.0.0) ; extra == "debug" or extra == "debug-server" or extra == "cli"
 Requires-Dist: sanic (>=20.12.2) ; extra == "sanic"
 Requires-Dist: starlette (>=0.18.0) ; extra == "asgi" or extra == "debug-server"
-Requires-Dist: starlite (>=1.48.0) ; (python_version >= "3.8") and (extra == "starlite")
 Requires-Dist: typer (>=0.7.0) ; extra == "debug-server" or extra == "cli"
 Requires-Dist: typing-extensions (>=4.5.0)
 Requires-Dist: uvicorn (>=0.11.6) ; extra == "debug-server"

{strawberry_graphql-0.237.3.dist-info → strawberry_graphql-0.238.0.dist-info}/RECORD

@@ -6,7 +6,7 @@ strawberry/aiohttp/handlers/graphql_transport_ws_handler.py,sha256=-dihpF3pueV6j
 strawberry/aiohttp/handlers/graphql_ws_handler.py,sha256=Ol8tBLIeal7bRiG_uxyJXnXo24xMDEh6M5XsuCyUpVQ,2275
 strawberry/aiohttp/test/__init__.py,sha256=4xxdUZtIISSOwjrcnmox7AvT4WWjowCm5bUuPdQneMg,71
 strawberry/aiohttp/test/client.py,sha256=4vjTDxtNVfpa74GfUUO7efPI6Ssh1vzfCYp3tPA-SLk,1357
-strawberry/aiohttp/views.py,sha256=3Imc-kP01zhDYmWQRFcQyZsObqdELQ84d-3zFrpHNxA,6091
+strawberry/aiohttp/views.py,sha256=q1y0zCuLNlwD_o4aosm7dNc-2F-ZJY95kiq1Ip0Prmw,6189
 strawberry/annotation.py,sha256=ftSxGZQUk4C5_5YRM_wNJFVVnZi0XOp71kD7zv4oxbU,13077
 strawberry/asgi/__init__.py,sha256=37w7FeDEmfq2YXKeyyBgjTAliz9ut5Y0-0dVgfE-PAk,6973
 strawberry/asgi/handlers/__init__.py,sha256=rz5Gth2eJUn7tDq2--99KSNFeMdDPpLFCkfA7vge0cI,235
@@ -114,7 +114,7 @@ strawberry/extensions/tracing/__init__.py,sha256=wx8_EAroGhNrP4HiGYMgKo8jnCsfde5
 strawberry/extensions/tracing/apollo.py,sha256=XlI88NzSZBSmBHEJ9iitDU9br2-9CdjdtHE_eiTJCIw,5880
 strawberry/extensions/tracing/datadog.py,sha256=khxvY4_WTjYaeJUb_dn6mLvmk9TCS4tIodnC3G-9pTo,5541
 strawberry/extensions/tracing/opentelemetry.py,sha256=MH2j71denfLmzInl6zcTzxzckqjcwOzm_F305z8-vYw,7281
-strawberry/extensions/tracing/sentry.py,sha256=y4km7GypJelFnlZbJ4L8aUR665T3sTIp4oauczA-uvE,5023
+strawberry/extensions/tracing/sentry.py,sha256=r_U1OeiDDq-Mf6v30-aEuHhrio2Pckxfrcrdu_OAnWM,5027
 strawberry/extensions/tracing/utils.py,sha256=tXZNyqfct6YNSWi3GRj4GU1fKQGvSce8ZESfoVeys7U,654
 strawberry/extensions/utils.py,sha256=YPiacKNLQXvpYj-HI6fR5ORFdM9RrcdabGeM7F8vBGg,1001
 strawberry/extensions/validation_cache.py,sha256=CZ-brPYA1grL_lW38Rq4TxEVKlHM2n1DC6q9BHlSF4g,1398
@@ -194,11 +194,6 @@ strawberry/schema/validation_rules/one_of.py,sha256=fPuYzCyLT7p9y7dHF_sWTImArTQa
 strawberry/schema_codegen/__init__.py,sha256=mhOfvC-h6ksmckXE2TC8WWlMgayNa2xwlbNelqFpeVY,24399
 strawberry/schema_directive.py,sha256=xFHoKOAWgVzDWt_N0ZdON3DBRByJkvMHt5TIvFLjUUU,2005
 strawberry/schema_directives.py,sha256=KGKFWCODjm1Ah9qNV_bBwbic7Mld4qLWnWQkev-PG8A,175
-strawberry/starlite/__init__.py,sha256=v209swT8H9MljVL-npvANhEO1zz3__PSfxb_Ix-NoeE,134
-strawberry/starlite/controller.py,sha256=ckZGcttZ-Ok30MSxtIiZO3wigJTw-it58s3kfHqua-s,12231
-strawberry/starlite/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/starlite/handlers/graphql_transport_ws_handler.py,sha256=k6kr8oTX5DGVpDHZkSotvYIcqKpyXO2qDr1RbKyiwXM,2023
-strawberry/starlite/handlers/graphql_ws_handler.py,sha256=BJtwhuIwaKfp8rGbvCBPKaDn_d2VY7Qd1iWdJhxHNK4,2256
 strawberry/static/apollo-sandbox.html,sha256=2XzkbE0dqsFHqehE-jul9_J9TFOpwA6Ylrlo0Kdx_9w,973
 strawberry/static/graphiql.html,sha256=r1JCF4vpptjISOAHaCrgck2b3qf0oaCEzljECWQ7aCE,4380
 strawberry/static/pathfinder.html,sha256=0DPx9AmJ2C_sJstFXnWOz9k5tVQHeHaK7qdVY4lAlmk,1547
@@ -248,8 +243,8 @@ strawberry/utils/logging.py,sha256=U1cseHGquN09YFhFmRkiphfASKCyK0HUZREImPgVb0c,7
 strawberry/utils/operation.py,sha256=SSXxN-vMqdHO6W2OZtip-1z7y4_A-eTVFdhDvhKeLCk,1193
 strawberry/utils/str_converters.py,sha256=KGd7QH90RevaJjH6SQEkiVVsb8KuhJr_wv5AsI7UzQk,897
 strawberry/utils/typing.py,sha256=tUHHX2YTGX417EEQHB6j0B8-p-fg31ZI8csc9SUoq2I,14260
-strawberry_graphql-0.237.3.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.237.3.dist-info/METADATA,sha256=sWyrEwqkg62uUtYwDNAq18skuwzvCaVZQnsLLCNeSKc,7821
-strawberry_graphql-0.237.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-strawberry_graphql-0.237.3.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.237.3.dist-info/RECORD,,
+strawberry_graphql-0.238.0.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.238.0.dist-info/METADATA,sha256=QyzfeHo1tGJ5zaJ3w5B7DcdL_4P1XXgZWNDjxyrnUYU,7707
+strawberry_graphql-0.238.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+strawberry_graphql-0.238.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.238.0.dist-info/RECORD,,

strawberry/starlite/__init__.py

@@ -1,3 +0,0 @@
-from strawberry.starlite.controller import BaseContext, make_graphql_controller
-
-__all__ = ["BaseContext", "make_graphql_controller"]

strawberry/starlite/controller.py

@@ -1,387 +0,0 @@
-"""Starlite integration for strawberry-graphql."""
-
-from __future__ import annotations
-
-import warnings
-from dataclasses import dataclass
-from datetime import timedelta
-from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, Tuple, Union, cast
-from typing_extensions import deprecated
-
-from starlite import (
-    BackgroundTasks,
-    Controller,
-    MediaType,
-    Provide,
-    Request,
-    Response,
-    WebSocket,
-    get,
-    post,
-    websocket,
-)
-from starlite.exceptions import (
-    NotFoundException,
-    ValidationException,
-)
-from starlite.status_codes import HTTP_200_OK
-from strawberry.exceptions import InvalidCustomContext
-from strawberry.http.async_base_view import AsyncBaseHTTPView, AsyncHTTPRequestAdapter
-from strawberry.http.exceptions import HTTPException
-from strawberry.http.types import FormData, HTTPMethod, QueryParams
-from strawberry.http.typevars import (
-    Context,
-    RootValue,
-)
-from strawberry.subscriptions import GRAPHQL_TRANSPORT_WS_PROTOCOL, GRAPHQL_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_transport_ws import (
-    WS_4406_PROTOCOL_NOT_ACCEPTABLE,
-)
-
-from .handlers.graphql_transport_ws_handler import (
-    GraphQLTransportWSHandler as BaseGraphQLTransportWSHandler,
-)
-from .handlers.graphql_ws_handler import GraphQLWSHandler as BaseGraphQLWSHandler
-
-if TYPE_CHECKING:
-    from typing import FrozenSet, List, Type
-
-    from starlite.types import AnyCallable, Dependencies
-    from strawberry.http import GraphQLHTTPResponse
-    from strawberry.http.ides import GraphQL_IDE
-    from strawberry.schema import BaseSchema
-
-    MergedContext = Union[
-        "BaseContext",
-        Union[
-            Dict[str, Any],
-            Dict[str, BackgroundTasks],
-            Dict[str, Request[Any, Any]],
-            Dict[str, Response[Any]],
-            Dict[str, websocket],
-        ],
-    ]
-
-CustomContext = Union["BaseContext", Dict[str, Any]]
-
-
-async def _context_getter(
-    custom_context: Optional[CustomContext],
-    request: Request[Any, Any],
-    response: Response[Any],
-) -> MergedContext:
-    if isinstance(custom_context, BaseContext):
-        custom_context.request = request
-        return custom_context
-
-    default_context = {
-        "request": request,
-        "response": response,
-    }
-
-    if isinstance(custom_context, dict):
-        return {
-            **default_context,
-            **custom_context,
-        }
-
-    if custom_context is None:
-        return default_context
-
-    raise InvalidCustomContext()
-
-
-@dataclass
-class GraphQLResource:
-    data: Optional[Dict[str, object]]
-    errors: Optional[List[object]]
-    extensions: Optional[Dict[str, object]]
-
-
-@dataclass
-class EmptyResponseModel:
-    pass
-
-
-class GraphQLWSHandler(BaseGraphQLWSHandler):
-    async def get_context(self) -> Any:
-        return await self._get_context()
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value()
-
-
-class GraphQLTransportWSHandler(BaseGraphQLTransportWSHandler):
-    async def get_context(self) -> Any:
-        return await self._get_context()
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value()
-
-
-@deprecated(
-    "The `starlite` integration is deprecated in favor of `litestar` integration",
-    stacklevel=2,
-)
-class StarliteRequestAdapter(AsyncHTTPRequestAdapter):
-    def __init__(self, request: Request[Any, Any]) -> None:
-        self.request = request
-
-    @property
-    def query_params(self) -> QueryParams:
-        return self.request.query_params
-
-    @property
-    def method(self) -> HTTPMethod:
-        return cast(HTTPMethod, self.request.method.upper())
-
-    @property
-    def headers(self) -> Mapping[str, str]:
-        return self.request.headers
-
-    @property
-    def content_type(self) -> Optional[str]:
-        return self.request.content_type[0]
-
-    async def get_body(self) -> bytes:
-        return await self.request.body()
-
-    async def get_form_data(self) -> FormData:
-        multipart_data = await self.request.form()
-
-        return FormData(form=multipart_data, files=multipart_data)
-
-
-class BaseContext:
-    def __init__(self) -> None:
-        self.request: Optional[Union[Request, WebSocket]] = None
-        self.response: Optional[Response] = None
-
-
-@deprecated(
-    "The `starlite` integration is deprecated in favor of `litestar` integration",
-    stacklevel=2,
-)
-def make_graphql_controller(
-    schema: BaseSchema,
-    path: str = "",
-    graphiql: Optional[bool] = None,
-    graphql_ide: Optional[GraphQL_IDE] = "graphiql",
-    allow_queries_via_get: bool = True,
-    keep_alive: bool = False,
-    keep_alive_interval: float = 1,
-    debug: bool = False,
-    # TODO: root typevar
-    root_value_getter: Optional[AnyCallable] = None,
-    # TODO: context typevar
-    context_getter: Optional[AnyCallable] = None,
-    subscription_protocols: Tuple[str, ...] = (
-        GRAPHQL_TRANSPORT_WS_PROTOCOL,
-        GRAPHQL_WS_PROTOCOL,
-    ),
-    connection_init_wait_timeout: timedelta = timedelta(minutes=1),
-) -> Type[Controller]:
-    routes_path = path
-
-    if context_getter is None:
-
-        def custom_context_getter_() -> None:
-            return None
-
-    else:
-        custom_context_getter_ = context_getter
-
-    if root_value_getter is None:
-
-        def root_value_getter_() -> None:
-            return None
-
-    else:
-        root_value_getter_ = root_value_getter
-
-    def response_getter() -> Response[Any]:
-        return Response({}, background=BackgroundTasks([]))
-
-    schema_ = schema
-    allow_queries_via_get_ = allow_queries_via_get
-    graphql_ide_: Optional[GraphQL_IDE]
-
-    if graphiql is not None:
-        warnings.warn(
-            "The `graphiql` argument is deprecated in favor of `graphql_ide`",
-            DeprecationWarning,
-            stacklevel=2,
-        )
-        graphql_ide_ = "graphiql" if graphiql else None
-    else:
-        graphql_ide_ = graphql_ide
-
-    class GraphQLController(
-        Controller,
-        AsyncBaseHTTPView[
-            Request[Any, Any], Response[Any], Response[Any], Context, RootValue
-        ],
-    ):
-        request_adapter_class = StarliteRequestAdapter
-        path: str = routes_path
-        dependencies: Optional[Dependencies] = {
-            "custom_context": Provide(custom_context_getter_),
-            "context": Provide(_context_getter),
-            "root_value": Provide(root_value_getter_),
-            "response": Provide(response_getter),
-        }
-        graphql_ws_handler_class: Type[GraphQLWSHandler] = GraphQLWSHandler
-        graphql_transport_ws_handler_class: Type[GraphQLTransportWSHandler] = (
-            GraphQLTransportWSHandler
-        )
-
-        _keep_alive: bool = keep_alive
-        _keep_alive_interval: float = keep_alive_interval
-        _debug: bool = debug
-        _protocols: Tuple[str, ...] = subscription_protocols
-        _connection_init_wait_timeout: timedelta = connection_init_wait_timeout
-        _graphiql_allowed_accept: FrozenSet[str] = frozenset({"text/html", "*/*"})
-
-        schema: BaseSchema = schema_
-        allow_queries_via_get = allow_queries_via_get_
-        graphql_ide = graphql_ide_
-
-        async def execute_request(
-            self,
-            request: Request[Any, Any],
-            context: CustomContext,
-            root_value: Any,
-        ) -> Response[Union[GraphQLResource, str]]:
-            try:
-                return await self.run(
-                    request,
-                    # TODO: check the dependency, above, can we make it so that
-                    # we don't need to type ignore here?
-                    context=context,  # type: ignore
-                    root_value=root_value,
-                )
-            except HTTPException as e:
-                return Response(
-                    e.reason,
-                    status_code=e.status_code,
-                    media_type=MediaType.TEXT,
-                )
-
-        async def render_graphql_ide(self, request: Request[Any, Any]) -> Response[str]:
-            return Response(self.graphql_ide_html, media_type=MediaType.HTML)
-
-        def create_response(
-            self, response_data: GraphQLHTTPResponse, sub_response: Response[bytes]
-        ) -> Response[bytes]:
-            response = Response(
-                self.encode_json(response_data).encode(),
-                status_code=HTTP_200_OK,
-                media_type=MediaType.JSON,
-            )
-
-            response.headers.update(sub_response.headers)
-            response.cookies.extend(sub_response.cookies)
-            response.background = sub_response.background
-
-            if sub_response.status_code:
-                response.status_code = sub_response.status_code
-
-            return response
-
-        @get(raises=[ValidationException, NotFoundException])
-        async def handle_http_get(
-            self,
-            request: Request[Any, Any],
-            context: CustomContext,
-            root_value: Any,
-            response: Response[Any],
-        ) -> Response[Union[GraphQLResource, str]]:
-            self.temporal_response = response
-
-            return await self.execute_request(
-                request=request,
-                context=context,
-                root_value=root_value,
-            )
-
-        @post(status_code=HTTP_200_OK)
-        async def handle_http_post(
-            self,
-            request: Request[Any, Any],
-            context: CustomContext,
-            root_value: Any,
-            response: Response[Any],
-        ) -> Response[Union[GraphQLResource, str]]:
-            self.temporal_response = response
-
-            return await self.execute_request(
-                request=request,
-                context=context,
-                root_value=root_value,
-            )
-
-        async def get_context(
-            self, request: Request[Any, Any], response: Response[Any]
-        ) -> Context:  # pragma: no cover
-            raise ValueError("`get_context` is not used by Starlite's controller")
-
-        async def get_root_value(
-            self, request: Request[Any, Any]
-        ) -> Optional[RootValue]:  # pragma: no cover
-            raise ValueError("`get_root_value` is not used by Starlite's controller")
-
-        async def get_sub_response(self, request: Request[Any, Any]) -> Response[Any]:
-            return self.temporal_response
-
-        @websocket()
-        async def websocket_endpoint(
-            self,
-            socket: WebSocket,
-            context: CustomContext,
-            root_value: Any,
-        ) -> None:
-            async def _get_context() -> CustomContext:
-                return context
-
-            async def _get_root_value() -> Any:
-                return root_value
-
-            preferred_protocol = self.pick_preferred_protocol(socket)
-            if preferred_protocol == GRAPHQL_TRANSPORT_WS_PROTOCOL:
-                await self.graphql_transport_ws_handler_class(
-                    schema=self.schema,
-                    debug=self._debug,
-                    connection_init_wait_timeout=self._connection_init_wait_timeout,
-                    get_context=_get_context,
-                    get_root_value=_get_root_value,
-                    ws=socket,
-                ).handle()
-            elif preferred_protocol == GRAPHQL_WS_PROTOCOL:
-                await self.graphql_ws_handler_class(
-                    schema=self.schema,
-                    debug=self._debug,
-                    keep_alive=self._keep_alive,
-                    keep_alive_interval=self._keep_alive_interval,
-                    get_context=_get_context,
-                    get_root_value=_get_root_value,
-                    ws=socket,
-                ).handle()
-            else:
-                await socket.close(code=WS_4406_PROTOCOL_NOT_ACCEPTABLE)
-
-        def pick_preferred_protocol(self, socket: WebSocket) -> Optional[str]:
-            protocols: List[str] = socket.scope["subprotocols"]
-            intersection = set(protocols) & set(self._protocols)
-            return (
-                min(
-                    intersection,
-                    key=lambda i: protocols.index(i) if i else "",
-                    default=None,
-                )
-                or None
-            )
-
-    return GraphQLController
-
-
-__all__ = ["make_graphql_controller"]

strawberry/starlite/handlers/graphql_transport_ws_handler.py

@@ -1,59 +0,0 @@
-from datetime import timedelta
-from typing import Any, Callable
-
-from starlite import WebSocket
-from starlite.exceptions import SerializationException, WebSocketDisconnect
-from strawberry.schema import BaseSchema
-from strawberry.subscriptions import GRAPHQL_TRANSPORT_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_transport_ws.handlers import (
-    BaseGraphQLTransportWSHandler,
-)
-
-
-class GraphQLTransportWSHandler(BaseGraphQLTransportWSHandler):
-    def __init__(
-        self,
-        schema: BaseSchema,
-        debug: bool,
-        connection_init_wait_timeout: timedelta,
-        get_context: Callable,
-        get_root_value: Callable,
-        ws: WebSocket,
-    ) -> None:
-        super().__init__(schema, debug, connection_init_wait_timeout)
-        self._get_context = get_context
-        self._get_root_value = get_root_value
-        self._ws = ws
-
-    async def get_context(self) -> Any:
-        return await self._get_context()
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value()
-
-    async def send_json(self, data: dict) -> None:
-        await self._ws.send_json(data)
-
-    async def close(self, code: int, reason: str) -> None:
-        await self._ws.close(code=code, reason=reason)
-
-    async def handle_request(self) -> None:
-        await self._ws.accept(subprotocols=GRAPHQL_TRANSPORT_WS_PROTOCOL)
-        self.on_request_accepted()
-
-        try:
-            while self._ws.connection_state != "disconnect":
-                try:
-                    message = await self._ws.receive_json()
-                except (SerializationException, ValueError):  # noqa: PERF203
-                    error_message = "WebSocket message type must be text"
-                    await self.handle_invalid_message(error_message)
-                else:
-                    await self.handle_message(message)
-        except WebSocketDisconnect:  # pragma: no cover
-            pass
-        finally:
-            await self.shutdown()
-
-
-__all__ = ["GraphQLTransportWSHandler"]

strawberry/starlite/handlers/graphql_ws_handler.py

@@ -1,64 +0,0 @@
-from contextlib import suppress
-from typing import Any, Callable, Optional
-
-from starlite import WebSocket
-from starlite.exceptions import SerializationException, WebSocketDisconnect
-from strawberry.schema import BaseSchema
-from strawberry.subscriptions import GRAPHQL_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_ws.handlers import BaseGraphQLWSHandler
-from strawberry.subscriptions.protocols.graphql_ws.types import OperationMessage
-
-
-class GraphQLWSHandler(BaseGraphQLWSHandler):
-    def __init__(
-        self,
-        schema: BaseSchema,
-        debug: bool,
-        keep_alive: bool,
-        keep_alive_interval: float,
-        get_context: Callable,
-        get_root_value: Callable,
-        ws: WebSocket,
-    ) -> None:
-        super().__init__(schema, debug, keep_alive, keep_alive_interval)
-        self._get_context = get_context
-        self._get_root_value = get_root_value
-        self._ws = ws
-
-    async def get_context(self) -> Any:
-        return await self._get_context()
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value()
-
-    async def send_json(self, data: OperationMessage) -> None:
-        await self._ws.send_json(data)
-
-    async def close(self, code: int = 1000, reason: Optional[str] = None) -> None:
-        await self._ws.close(code=code, reason=reason)
-
-    async def handle_request(self) -> Any:
-        await self._ws.accept(subprotocols=GRAPHQL_WS_PROTOCOL)
-
-        try:
-            while self._ws.connection_state != "disconnect":
-                try:
-                    message = await self._ws.receive_json()
-                except (SerializationException, ValueError):  # noqa: PERF203
-                    # Ignore non-text messages
-                    continue
-                else:
-                    await self.handle_message(message)
-        except WebSocketDisconnect:  # pragma: no cover
-            pass
-        finally:
-            if self.keep_alive_task:
-                self.keep_alive_task.cancel()
-                with suppress(BaseException):
-                    await self.keep_alive_task
-
-            for operation_id in list(self.subscriptions.keys()):
-                await self.cleanup_operation(operation_id)
-
-
-__all__ = ["GraphQLWSHandler"]