strawberry-graphql 0.243.1__py3-none-any.whl → 0.244.1__py3-none-any.whl

strawberry/subscriptions/protocols/graphql_ws/handlers.py

@@ -1,11 +1,9 @@
 from __future__ import annotations
 
 import asyncio
-from abc import ABC, abstractmethod
 from contextlib import suppress
 from typing import (
     TYPE_CHECKING,
-    Any,
     AsyncGenerator,
     Awaitable,
     Dict,
@@ -13,6 +11,7 @@ from typing import (
     cast,
 )
 
+from strawberry.http.exceptions import NonJsonMessageReceived
 from strawberry.subscriptions.protocols.graphql_ws import (
     GQL_COMPLETE,
     GQL_CONNECTION_ACK,
@@ -25,29 +24,36 @@ from strawberry.subscriptions.protocols.graphql_ws import (
     GQL_START,
     GQL_STOP,
 )
+from strawberry.subscriptions.protocols.graphql_ws.types import (
+    ConnectionInitPayload,
+    DataPayload,
+    OperationMessage,
+    OperationMessagePayload,
+    StartPayload,
+)
 from strawberry.types.execution import ExecutionResult, PreExecutionError
 from strawberry.utils.debug import pretty_print_graphql_operation
 
 if TYPE_CHECKING:
+    from strawberry.http.async_base_view import AsyncWebSocketAdapter
     from strawberry.schema import BaseSchema
     from strawberry.schema.subscribe import SubscriptionResult
-    from strawberry.subscriptions.protocols.graphql_ws.types import (
-        ConnectionInitPayload,
-        DataPayload,
-        OperationMessage,
-        OperationMessagePayload,
-        StartPayload,
-    )
 
 
-class BaseGraphQLWSHandler(ABC):
+class BaseGraphQLWSHandler:
     def __init__(
         self,
+        websocket: AsyncWebSocketAdapter,
+        context: object,
+        root_value: object,
         schema: BaseSchema,
         debug: bool,
         keep_alive: bool,
-        keep_alive_interval: float,
+        keep_alive_interval: Optional[float],
     ) -> None:
+        self.websocket = websocket
+        self.context = context
+        self.root_value = root_value
         self.schema = schema
         self.debug = debug
         self.keep_alive = keep_alive
@@ -57,28 +63,22 @@ class BaseGraphQLWSHandler(ABC):
         self.tasks: Dict[str, asyncio.Task] = {}
         self.connection_params: Optional[ConnectionInitPayload] = None
 
-    @abstractmethod
-    async def get_context(self) -> Any:
-        """Return the operations context."""
-
-    @abstractmethod
-    async def get_root_value(self) -> Any:
-        """Return the schemas root value."""
-
-    @abstractmethod
-    async def send_json(self, data: OperationMessage) -> None:
-        """Send the data JSON encoded to the WebSocket client."""
-
-    @abstractmethod
-    async def close(self, code: int = 1000, reason: Optional[str] = None) -> None:
-        """Close the WebSocket with the passed code and reason."""
-
-    @abstractmethod
-    async def handle_request(self) -> Any:
-        """Handle the request this instance was created for."""
-
-    async def handle(self) -> Any:
-        return await self.handle_request()
+    async def handle(self) -> None:
+        try:
+            async for message in self.websocket.iter_json():
+                await self.handle_message(cast(OperationMessage, message))
+        except NonJsonMessageReceived:
+            await self.websocket.close(
+                code=1002, reason="WebSocket message type must be text"
+            )
+        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)
 
     async def handle_message(
         self,
@@ -99,22 +99,22 @@ class BaseGraphQLWSHandler(ABC):
         payload = message.get("payload")
         if payload is not None and not isinstance(payload, dict):
             error_message: OperationMessage = {"type": GQL_CONNECTION_ERROR}
-            await self.send_json(error_message)
-            await self.close()
+            await self.websocket.send_json(error_message)
+            await self.websocket.close(code=1000, reason="")
             return
 
         payload = cast(Optional["ConnectionInitPayload"], payload)
         self.connection_params = payload
 
         acknowledge_message: OperationMessage = {"type": GQL_CONNECTION_ACK}
-        await self.send_json(acknowledge_message)
+        await self.websocket.send_json(acknowledge_message)
 
         if self.keep_alive:
             keep_alive_handler = self.handle_keep_alive()
             self.keep_alive_task = asyncio.create_task(keep_alive_handler)
 
     async def handle_connection_terminate(self, message: OperationMessage) -> None:
-        await self.close()
+        await self.websocket.close(code=1000, reason="")
 
     async def handle_start(self, message: OperationMessage) -> None:
         operation_id = message["id"]
@@ -123,10 +123,10 @@ class BaseGraphQLWSHandler(ABC):
         operation_name = payload.get("operationName")
         variables = payload.get("variables")
 
-        context = await self.get_context()
-        if isinstance(context, dict):
-            context["connection_params"] = self.connection_params
-        root_value = await self.get_root_value()
+        if isinstance(self.context, dict):
+            self.context["connection_params"] = self.connection_params
+        elif hasattr(self.context, "connection_params"):
+            self.context.connection_params = self.connection_params
 
         if self.debug:
             pretty_print_graphql_operation(operation_name, query, variables)
@@ -135,8 +135,8 @@ class BaseGraphQLWSHandler(ABC):
             query=query,
             variable_values=variables,
             operation_name=operation_name,
-            context_value=context,
-            root_value=root_value,
+            context_value=self.context,
+            root_value=self.root_value,
         )
 
         result_handler = self.handle_async_results(result_source, operation_id)
@@ -147,9 +147,10 @@ class BaseGraphQLWSHandler(ABC):
         await self.cleanup_operation(operation_id)
 
     async def handle_keep_alive(self) -> None:
+        assert self.keep_alive_interval
         while True:
             data: OperationMessage = {"type": GQL_CONNECTION_KEEP_ALIVE}
-            await self.send_json(data)
+            await self.websocket.send_json(data)
             await asyncio.sleep(self.keep_alive_interval)
 
     async def handle_async_results(
@@ -191,7 +192,7 @@ class BaseGraphQLWSHandler(ABC):
         data: OperationMessage = {"type": type_, "id": operation_id}
         if payload is not None:
             data["payload"] = payload
-        await self.send_json(data)
+        await self.websocket.send_json(data)
 
     async def send_data(
         self, execution_result: ExecutionResult, operation_id: str

{strawberry_graphql-0.243.1.dist-info → strawberry_graphql-0.244.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strawberry-graphql
-Version: 0.243.1
+Version: 0.244.1
 Summary: A library for creating GraphQL APIs
 Home-page: https://strawberry.rocks/
 License: MIT

{strawberry_graphql-0.243.1.dist-info → strawberry_graphql-0.244.1.dist-info}/RECORD

@@ -1,28 +1,20 @@
 strawberry/__init__.py,sha256=tJOqxIEg3DYPe48P952sMjF3IxxZqZfIY-kNRlbwBNg,1374
 strawberry/__main__.py,sha256=3U77Eu21mJ-LY27RG-JEnpbh6Z63wGOom4i-EoLtUcY,59
 strawberry/aiohttp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/aiohttp/handlers/__init__.py,sha256=7EeGIIwrgJYHAS9XJnZLfRGL_GHrligKm39rrVt7qDA,241
-strawberry/aiohttp/handlers/graphql_transport_ws_handler.py,sha256=-dihpF3pueV6jcd0x4k0GF-3VtG_IvfyKQ1ZVgg1T4U,2092
-strawberry/aiohttp/handlers/graphql_ws_handler.py,sha256=3U-E-o_QcLGoVaHE2oAPN3tTTIRwqGH0_688kbOtT7I,2437
 strawberry/aiohttp/test/__init__.py,sha256=4xxdUZtIISSOwjrcnmox7AvT4WWjowCm5bUuPdQneMg,71
 strawberry/aiohttp/test/client.py,sha256=4vjTDxtNVfpa74GfUUO7efPI6Ssh1vzfCYp3tPA-SLk,1357
-strawberry/aiohttp/views.py,sha256=z9T3hJU9m410VjcZlxiatirh04-4-bPSzW7eCGqO0dc,7052
+strawberry/aiohttp/views.py,sha256=wbJkcQ1u98tuHa3_mq8OCGzSm57Nnw4DnS5qkILHlEo,7487
 strawberry/annotation.py,sha256=ftSxGZQUk4C5_5YRM_wNJFVVnZi0XOp71kD7zv4oxbU,13077
-strawberry/asgi/__init__.py,sha256=7YmrcQuD7fEPWo1Q60GVvVZdvMCNrTaic1y4pA_NpF4,7642
-strawberry/asgi/handlers/__init__.py,sha256=rz5Gth2eJUn7tDq2--99KSNFeMdDPpLFCkfA7vge0cI,235
-strawberry/asgi/handlers/graphql_transport_ws_handler.py,sha256=_5N58XdtCZndU81ky1f5cwG9E4NhysxP4uHlqqZNzn4,2147
-strawberry/asgi/handlers/graphql_ws_handler.py,sha256=UnjkjyEen2P4Ld9AuazcZ2qY-VRdBubunb4JHtAP89o,2443
+strawberry/asgi/__init__.py,sha256=aFM74LpXxTruW5a00Ry9rIi3vz7hgnELJvZqPwJnhcY,7645
 strawberry/asgi/test/__init__.py,sha256=4xxdUZtIISSOwjrcnmox7AvT4WWjowCm5bUuPdQneMg,71
 strawberry/asgi/test/client.py,sha256=VolupxMna9ktF1lYgV_dUQAIN53DNzVyWTeWTbwsxqE,1448
 strawberry/chalice/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/chalice/views.py,sha256=VXXOfR3JIEf5CsPMZe0sXBnsH097uPyPe_CdqwESVv4,4770
-strawberry/channels/__init__.py,sha256=9oRdAT7uIYETF-23gZ5NteOXSjwkUtwRmwu3YCFv1tw,671
+strawberry/channels/__init__.py,sha256=-9ENTIu1AILbqffJ663qH6AwpZgLrJx_kaokS7RrCnA,433
 strawberry/channels/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/channels/handlers/base.py,sha256=DlMmzuZl-feceZZ7UHLW-zBSNVxHvaWTqwyWDP-e1Dg,7861
-strawberry/channels/handlers/graphql_transport_ws_handler.py,sha256=jMWmZ4Haoo3bf_bcM4ZheAjkreuhZBhwMhEX6GZ0Ue4,1995
-strawberry/channels/handlers/graphql_ws_handler.py,sha256=ulkMQP7h5cFDDLVBWV-10fYa9GS2pFg3FdS0Axtov-U,2583
-strawberry/channels/handlers/http_handler.py,sha256=KMvbsU6aYGkOnBVLdhWNIv5HJxD41qdx9OkFZnn8m0I,11258
-strawberry/channels/handlers/ws_handler.py,sha256=hpgXSWsUMWP3ylJtHGFcLtNpSKgPMWJTfnUTKElBBzA,4747
+strawberry/channels/handlers/base.py,sha256=KV4KA0eF5NRtikV9m4ssoPI5pmCZvDuRkfoTxwh42qI,7853
+strawberry/channels/handlers/http_handler.py,sha256=HPxAddsVPaPS293CTvgYaxS8ot32B2t_BqSJPtywVjQ,11744
+strawberry/channels/handlers/ws_handler.py,sha256=mkfYKtGQO7So4BVBcUHQqG0vUVWIx55vHbJq-D5kz5Y,5924
 strawberry/channels/router.py,sha256=DKIbl4zuRBhfvViUVpyu0Rf_WRT41E6uZC-Yic9Ltvo,2024
 strawberry/channels/testing.py,sha256=GZqYu_rhrT1gLHmdI219L1fctVDmrv7AMHs0bwhXitc,6166
 strawberry/cli/__init__.py,sha256=byS5VrEiTJatAH6YS4V1Kd4SOwMRAQO2M1oJdIddivg,585
@@ -45,7 +37,7 @@ strawberry/codegen/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 strawberry/codegen/plugins/print_operation.py,sha256=aw2o-41nZSZ2eOHPyp58-45QsAgUs1MmnHDIQzQRg9M,6805
 strawberry/codegen/plugins/python.py,sha256=aellTnJAZOfTZqFXH4fEGYFjnnCucyroeOXo9D0QkrA,6924
 strawberry/codegen/plugins/typescript.py,sha256=f3JJ0FGuhycu-bxOkW7KxAyYdfmcbej-lueOJZx3V7Y,3783
-strawberry/codegen/query_codegen.py,sha256=Fc9oDprjXPV1OlIPmHUgI7iW6R_dhUj7QBTFM0cTsqc,30363
+strawberry/codegen/query_codegen.py,sha256=rKxseU-XiAUnQDX3V8ZAiWtyigVF_CBi2Pvvkomnqtk,30497
 strawberry/codegen/types.py,sha256=TBVjaKHBvr9QwELRdZUVlWS01wIEVXXTWs5etjqHVhk,4162
 strawberry/codemods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/codemods/annotated_unions.py,sha256=TStU8ye5J1---P5mNWXBGeQa8m_J7FyPU_MYjzDZ-RU,5910
@@ -57,7 +49,7 @@ strawberry/django/apps.py,sha256=ZWw3Mzv1Cgy0T9xT8Jr2_dkCTZjT5WQABb34iqnu5pc,135
 strawberry/django/context.py,sha256=XL85jDGAVnb2pwgm5uRUvIXwlGia3i-8ZVfKihf0T24,655
 strawberry/django/test/__init__.py,sha256=4xxdUZtIISSOwjrcnmox7AvT4WWjowCm5bUuPdQneMg,71
 strawberry/django/test/client.py,sha256=6dorWECd0wdn8fu3dabE-dfGK3uza58mGrdJ-xPct-w,626
-strawberry/django/views.py,sha256=Is20K6DVqClY7HSpjgoc668qL5HlxZBpMnNYz0DDpBk,9810
+strawberry/django/views.py,sha256=OMjjUFdlQeDUsF89cqmxU-k1e5kYp0omVoFgu96KjBE,10332
 strawberry/exceptions/__init__.py,sha256=DgdOJUs2xXHWcakr4tN6iIogltPi0MNnpu6MM6K0p5k,6347
 strawberry/exceptions/conflicting_arguments.py,sha256=68f6kMSXdjuEjZkoe8o2I9PSIjwTS1kXsSGaQBPk_hI,1587
 strawberry/exceptions/duplicated_type_name.py,sha256=-FG5qG_Mvkd7ROdOxCB9bijf8QR6Olryf07mbAFC0-U,2210
@@ -120,10 +112,7 @@ strawberry/extensions/utils.py,sha256=YPiacKNLQXvpYj-HI6fR5ORFdM9RrcdabGeM7F8vBG
 strawberry/extensions/validation_cache.py,sha256=CZ-brPYA1grL_lW38Rq4TxEVKlHM2n1DC6q9BHlSF4g,1398
 strawberry/fastapi/__init__.py,sha256=p5qg9AlkYjNOWKcT4uRiebIpR6pIb1HqDMiDfF5O3tg,147
 strawberry/fastapi/context.py,sha256=07991DShQoYMBVyQl9Mh5xvXQSNQI2RXT2ZQ5fWiga4,690
-strawberry/fastapi/handlers/__init__.py,sha256=TziBHyibYBOGiidjpCkjNThYbVY7K_nt0hnRLVHVh3I,241
-strawberry/fastapi/handlers/graphql_transport_ws_handler.py,sha256=5ivH7DJup0ZyGawAcj-n9VE_exBTje9Hou0_GIZCyD4,591
-strawberry/fastapi/handlers/graphql_ws_handler.py,sha256=OTloVUvEgXDpqjOlBAT1FnaNWRAglQgAt613YH06roc,537
-strawberry/fastapi/router.py,sha256=B2_FjFAD27_yZKR2PoMUNfW7If9huPyzd5f-Bwbz6ZI,13054
+strawberry/fastapi/router.py,sha256=XdVgpNKyUfwNiqVSheXyWTNoODIbyxXY3imrLBZ_bYU,12242
 strawberry/federation/__init__.py,sha256=FeUxLiBVuk9TKBmJJi51SeUaI8c80rS8hbl9No-hjII,535
 strawberry/federation/argument.py,sha256=2m_wgp7uFQDimTDDCBBqSoWeTop4MD1-KjjYrumEJIw,833
 strawberry/federation/enum.py,sha256=1rx50FiSMS-NjEFErhRSYB5no8TPsGMA_cH7hVbxAFI,3015
@@ -142,22 +131,19 @@ strawberry/file_uploads/__init__.py,sha256=v2-6FGBqnTnMPSUTFOiXpIutDMl-ga0PFtw5t
 strawberry/file_uploads/scalars.py,sha256=NRDeB7j8aotqIkz9r62ISTf4DrxQxEZYUuHsX5K16aU,161
 strawberry/file_uploads/utils.py,sha256=2zsXg3QsKgGLD7of2dW-vgQn_Naf7I3Men9PhEAFYwM,1160
 strawberry/flask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/flask/views.py,sha256=Tc9Y1WwQyXR90-F2MEPpNkIcxuBri5iYCG69OnpbSe8,5772
+strawberry/flask/views.py,sha256=dbYDuZKOiC7fFKLtU1x32utgfi6SLqJtAqPtDvUxcq4,6215
 strawberry/http/__init__.py,sha256=GSvHUDXl1cHfLnb37PXOAnxfoXhvz0f467P1O8uDatM,1620
-strawberry/http/async_base_view.py,sha256=moI28qAGNqoKgWq5ZfmgNxmY97E8UF_Wk3PYFPeVh-g,11805
+strawberry/http/async_base_view.py,sha256=ZWxxLNCISinp9nc7878_0FjRt8y-giQbtJgSF-Q4ZF0,15688
 strawberry/http/base.py,sha256=swbBPIl1SYwX7gPq25ad4SyXg2Nl7ccM-0CrLFGT_FI,2613
-strawberry/http/exceptions.py,sha256=WdWO3RvZDax_yAdD0zlVex9tQgwNx7tjz8_A8kP4YHo,193
+strawberry/http/exceptions.py,sha256=-mRC6RALAj3XCaKeYvHNDiIRAMbE_GbVd7TEAgqRocA,245
 strawberry/http/ides.py,sha256=3dqFRY8_9ZqyIYR_EyRdPZ1zhL3lxRYT2MPk84O_Tk8,874
 strawberry/http/parse_content_type.py,sha256=sgtcOO_ZOFg7WWWibYyLc4SU58K-SErcW56kQczQmKU,412
 strawberry/http/sync_base_view.py,sha256=qA9o-Ic4ZcTXiKF02lBsrN7ET6VeXGYWf9m9mjhlfWU,7199
 strawberry/http/temporal_response.py,sha256=QrGYSg7Apu7Mh-X_uPKDZby-UicXw2J_ywxaqhS8a_4,220
 strawberry/http/types.py,sha256=cAuaiUuvaMI_XhZ2Ey6Ej23WyQKqMGFxzzpVHDjVazY,371
-strawberry/http/typevars.py,sha256=flx5KPWnTwYju7VwRSVhMmx15Rl1pQT1K57_GnK72Hg,282
+strawberry/http/typevars.py,sha256=8hK5PfNPZXb2EhZmqlobYyfwJJcO2Wb96T91MlLEVJs,450
 strawberry/litestar/__init__.py,sha256=zsXzg-mglCGUVO9iNXLm-yadoDSCK7k-zuyRqyvAh1w,237
-strawberry/litestar/controller.py,sha256=0t4BZ69sBYYndEqbLVm7TP7gLYSdY1tfJnYI7mccE9A,14184
-strawberry/litestar/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/litestar/handlers/graphql_transport_ws_handler.py,sha256=rDE-I02_fCov4FfpdBJBE2Xt-4FSNU8xJuQ6xedMF6E,2050
-strawberry/litestar/handlers/graphql_ws_handler.py,sha256=b4gz9aEKtpXUcdyB3o9PzmINcuNReIshJM1bfPXH14Q,2347
+strawberry/litestar/controller.py,sha256=oKjio86UsFjvdwxx80YqX_-WWnw51Sd7saMulQ0ihIQ,13519
 strawberry/parent.py,sha256=sXURm0lauSpjUADsmfNGY-Zl7kHs0A67BFcWuWKzRxw,771
 strawberry/permission.py,sha256=NsYq-c4AgDCDBsYXsJN94yWJjuwYkvytpqXE4BIf_vc,7226
 strawberry/printer/__init__.py,sha256=DmepjmgtkdF5RxK_7yC6qUyRWn56U-9qeZMbkztYB9w,62
@@ -165,7 +151,7 @@ strawberry/printer/ast_from_value.py,sha256=LgM5g2qvBOnAIf9znbiMEcRX0PGSQohR3Vr3
 strawberry/printer/printer.py,sha256=GntTBivg3fb_zPM41Q8DtWMiRmkmM9xwTF-aFWvnqTg,17524
 strawberry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/quart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-strawberry/quart/views.py,sha256=wuc5ESBPNKhY8QliZDPmOzk2WsdUvItnSv_-vVPBcLU,4002
+strawberry/quart/views.py,sha256=-yjbFlXN1CLJqPrAgynGK_WwATTqsfBF-Kh_bJQ4JYs,4445
 strawberry/relay/__init__.py,sha256=Vi4btvA_g6Cj9Tk_F9GCSegapIf2WqkOWV8y3P0cTCs,553
 strawberry/relay/exceptions.py,sha256=KZSRJYlfutrAQALtBPnzJHRIMK6GZSnKAT_H4wIzGcI,4035
 strawberry/relay/fields.py,sha256=qrpDxDQ_bdzDUKtd8gxo9P3Oermxbux3yjFvHvHC1Ho,16927
@@ -175,7 +161,7 @@ strawberry/resolvers.py,sha256=Vdidc3YFc4-olSQZD_xQ1icyAFbyzqs_8I3eSpMFlA4,260
 strawberry/sanic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/sanic/context.py,sha256=qN7I9K_qIqgdbG_FbDl8XMb9aM1PyjIxSo8IAg2Uq8o,844
 strawberry/sanic/utils.py,sha256=r-tCX0JzELAdupF7fFy65JWAxj6pABSntNbGNaGZT8o,1080
-strawberry/sanic/views.py,sha256=YfTDowINXBz8P_zHH6-u1_agCG1okbWAURdvP4HG3eQ,6581
+strawberry/sanic/views.py,sha256=O4w67pDn0ecefcbbuytCfvZARI4IbXhTPfRzo6MLMQg,7089
 strawberry/scalars.py,sha256=c3y8EOmX-KUxSgRqk1TNercMA6_rgBHhQPp0z3C2zBU,2240
 strawberry/schema/__init__.py,sha256=u1QCyDVQExUVDA20kyosKPz3TS5HMCN2NrXclhiFAL4,92
 strawberry/schema/base.py,sha256=yarj62fhfCp0kToJPpWlNcCjyZV2CafbfpZ-yamjNVg,3730
@@ -202,10 +188,10 @@ strawberry/static/pathfinder.html,sha256=0DPx9AmJ2C_sJstFXnWOz9k5tVQHeHaK7qdVY4l
 strawberry/subscriptions/__init__.py,sha256=1VGmiCzFepqRFyCikagkUoHHdoTG3XYlFu9GafoQMws,170
 strawberry/subscriptions/protocols/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/subscriptions/protocols/graphql_transport_ws/__init__.py,sha256=wN6dkMu6WiaIZTE19PGoN9xXpIN_RdDE_q7F7ZgjCxk,138
-strawberry/subscriptions/protocols/graphql_transport_ws/handlers.py,sha256=qTz1yN0B9KRl01MZPOwgCvli5S9OIG2y7BJLGDLJHgA,14219
+strawberry/subscriptions/protocols/graphql_transport_ws/handlers.py,sha256=oh1XUlhl3lZx0bAU1Ta7cC9f9nuDbJGFnxRlm1z6ud8,14388
 strawberry/subscriptions/protocols/graphql_transport_ws/types.py,sha256=udYxzGtwjETYvY5f23org0t-aY4cimTjEGFYUR3idaY,2596
 strawberry/subscriptions/protocols/graphql_ws/__init__.py,sha256=ijn1A1O0Fzv5p9n-jw3T5H7M3oxbN4gbxRaepN7HyJk,553
-strawberry/subscriptions/protocols/graphql_ws/handlers.py,sha256=UfCvmVve26HzhixP-CeKHEjvQGsAx5_iizt_Mww2uNc,7175
+strawberry/subscriptions/protocols/graphql_ws/handlers.py,sha256=QxCribyj-EP9SFblbUd4GxK4dMdJe-P8TsjYPwaXf-8,7556
 strawberry/subscriptions/protocols/graphql_ws/types.py,sha256=CKm4Hy95p6H9u_-QyWdxipwzNeeDIWtilP3RRp8vjPw,1050
 strawberry/test/__init__.py,sha256=U3B5Ng7C_H8GpCpfvgZZcfADMw6cor5hm78gS3nDdMI,115
 strawberry/test/client.py,sha256=-V_5DakR0NJ_Kd5ww4leIhMnIJ-Pf274HkNqYtGVfl8,6044
@@ -245,8 +231,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=3xws5kxSQGsp8BnYyUwClvxXNzZakMAuOPoq1rjHRuk,14252
-strawberry_graphql-0.243.1.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.243.1.dist-info/METADATA,sha256=pUMQj5rzTkjqWbNbKuzGnpZWoNaWKyyhKnkyKwzsvx0,7707
-strawberry_graphql-0.243.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-strawberry_graphql-0.243.1.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.243.1.dist-info/RECORD,,
+strawberry_graphql-0.244.1.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.244.1.dist-info/METADATA,sha256=sIPmwJwtaUc8pbqGxSr-8WOYHOMRzTEGXM8nPfedFo8,7707
+strawberry_graphql-0.244.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+strawberry_graphql-0.244.1.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.244.1.dist-info/RECORD,,

strawberry/aiohttp/handlers/__init__.py

@@ -1,6 +0,0 @@
-from strawberry.aiohttp.handlers.graphql_transport_ws_handler import (
-    GraphQLTransportWSHandler,
-)
-from strawberry.aiohttp.handlers.graphql_ws_handler import GraphQLWSHandler
-
-__all__ = ["GraphQLTransportWSHandler", "GraphQLWSHandler"]

strawberry/aiohttp/handlers/graphql_transport_ws_handler.py

@@ -1,62 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING, Any, Callable, Dict
-
-from aiohttp import http, web
-from strawberry.subscriptions import GRAPHQL_TRANSPORT_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_transport_ws.handlers import (
-    BaseGraphQLTransportWSHandler,
-)
-
-if TYPE_CHECKING:
-    from datetime import timedelta
-
-    from strawberry.schema import BaseSchema
-
-
-class GraphQLTransportWSHandler(BaseGraphQLTransportWSHandler):
-    def __init__(
-        self,
-        schema: BaseSchema,
-        debug: bool,
-        connection_init_wait_timeout: timedelta,
-        get_context: Callable[..., Dict[str, Any]],
-        get_root_value: Any,
-        request: web.Request,
-    ) -> None:
-        super().__init__(schema, debug, connection_init_wait_timeout)
-        self._get_context = get_context
-        self._get_root_value = get_root_value
-        self._request = request
-        self._ws = web.WebSocketResponse(protocols=[GRAPHQL_TRANSPORT_WS_PROTOCOL])
-
-    async def get_context(self) -> Any:
-        return await self._get_context(request=self._request, response=self._ws)  # type: ignore
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value(request=self._request)
-
-    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, message=reason.encode())
-
-    async def handle_request(self) -> web.StreamResponse:
-        await self._ws.prepare(self._request)
-        self.on_request_accepted()
-
-        try:
-            async for ws_message in self._ws:  # type: http.WSMessage
-                if ws_message.type == http.WSMsgType.TEXT:
-                    await self.handle_message(ws_message.json())
-                else:
-                    error_message = "WebSocket message type must be text"
-                    await self.handle_invalid_message(error_message)
-        finally:
-            await self.shutdown()
-
-        return self._ws
-
-
-__all__ = ["GraphQLTransportWSHandler"]

strawberry/aiohttp/handlers/graphql_ws_handler.py

@@ -1,69 +0,0 @@
-from __future__ import annotations
-
-from contextlib import suppress
-from typing import TYPE_CHECKING, Any, Callable, Optional
-
-from aiohttp import http, web
-from strawberry.subscriptions import GRAPHQL_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_ws.handlers import BaseGraphQLWSHandler
-
-if TYPE_CHECKING:
-    from strawberry.schema import BaseSchema
-    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,
-        request: web.Request,
-    ) -> None:
-        super().__init__(schema, debug, keep_alive, keep_alive_interval)
-        self._get_context = get_context
-        self._get_root_value = get_root_value
-        self._request = request
-        self._ws = web.WebSocketResponse(protocols=[GRAPHQL_WS_PROTOCOL])
-
-    async def get_context(self) -> Any:
-        return await self._get_context(request=self._request, response=self._ws)
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value(request=self._request)
-
-    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:
-        message = reason.encode() if reason else b""
-        await self._ws.close(code=code, message=message)
-
-    async def handle_request(self) -> Any:
-        await self._ws.prepare(self._request)
-
-        try:
-            async for ws_message in self._ws:  # type: http.WSMessage
-                if ws_message.type == http.WSMsgType.TEXT:
-                    message: OperationMessage = ws_message.json()
-                    await self.handle_message(message)
-                else:
-                    await self.close(
-                        code=1002, reason="WebSocket message type must be text"
-                    )
-        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)
-
-        return self._ws
-
-
-__all__ = ["GraphQLWSHandler"]

strawberry/asgi/handlers/__init__.py

@@ -1,6 +0,0 @@
-from strawberry.asgi.handlers.graphql_transport_ws_handler import (
-    GraphQLTransportWSHandler,
-)
-from strawberry.asgi.handlers.graphql_ws_handler import GraphQLWSHandler
-
-__all__ = ["GraphQLTransportWSHandler", "GraphQLWSHandler"]

strawberry/asgi/handlers/graphql_transport_ws_handler.py

@@ -1,66 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING, Any, Callable
-
-from starlette.websockets import WebSocketDisconnect, WebSocketState
-
-from strawberry.subscriptions import GRAPHQL_TRANSPORT_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_transport_ws.handlers import (
-    BaseGraphQLTransportWSHandler,
-)
-
-if TYPE_CHECKING:
-    from datetime import timedelta
-
-    from starlette.websockets import WebSocket
-
-    from strawberry.schema import BaseSchema
-
-
-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(request=self._ws, response=None)
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value(request=self._ws)
-
-    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(subprotocol=GRAPHQL_TRANSPORT_WS_PROTOCOL)
-        self.on_request_accepted()
-
-        try:
-            while self._ws.application_state != WebSocketState.DISCONNECTED:
-                try:
-                    message = await self._ws.receive_json()
-                except KeyError:  # 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/asgi/handlers/graphql_ws_handler.py

@@ -1,71 +0,0 @@
-from __future__ import annotations
-
-from contextlib import suppress
-from typing import TYPE_CHECKING, Any, Callable, Optional
-
-from starlette.websockets import WebSocketDisconnect, WebSocketState
-
-from strawberry.subscriptions import GRAPHQL_WS_PROTOCOL
-from strawberry.subscriptions.protocols.graphql_ws.handlers import BaseGraphQLWSHandler
-
-if TYPE_CHECKING:
-    from starlette.websockets import WebSocket
-
-    from strawberry.schema import BaseSchema
-    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(request=self._ws, response=None)
-
-    async def get_root_value(self) -> Any:
-        return await self._get_root_value(request=self._ws)
-
-    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(subprotocol=GRAPHQL_WS_PROTOCOL)
-
-        try:
-            while self._ws.application_state != WebSocketState.DISCONNECTED:
-                try:
-                    message = await self._ws.receive_json()
-                except KeyError:  # noqa: PERF203
-                    await self.close(
-                        code=1002, reason="WebSocket message type must be text"
-                    )
-                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"]