strawberry-graphql 0.217.0__tar.gz → 0.218.0.dev1705418681__tar.gz

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/PKG-INFO

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

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "strawberry-graphql"
 packages = [ { include = "strawberry" } ]
-version = "0.217.0"
+version = "0.218.0.dev.1705418681"
 description = "A library for creating GraphQL APIs"
 authors = ["Patrick Arminio <patrick.arminio@gmail.com>"]
 license = "MIT"
@@ -75,7 +75,7 @@ opentelemetry-sdk = "<2"
 pygments = "^2.3"
 pyinstrument = {version = ">=4.0.0", optional = false}
 pytest = "^7.2"
-pytest-asyncio = ">=0.20.3,<0.22.0"
+pytest-asyncio = ">=0.20.3,<0.24.0"
 pytest-codspeed = "^2.0.1"
 pytest-cov = "^4.0.0"
 pytest-emoji = "^0.2.0"
@@ -96,8 +96,8 @@ types-toml = "^0.10.8"
 types-typed-ast = "^1.5.8"
 types-ujson = "^5.6.0"
 # added this here manually because poetry doesn't seem to be able to handle it =(
-botocore = "1.31.84"
-mypy = "1.7.0"
+botocore = "1.34.13"
+mypy = "1.8.0"
 pytest-mypy-plugins = ">=1.10,<4.0"
 types-protobuf = "^4.23.0.1"
 

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/aiohttp/views.py

@@ -7,10 +7,13 @@ from io import BytesIO
 from typing import (
     TYPE_CHECKING,
     Any,
+    AsyncGenerator,
+    Callable,
     Dict,
     Iterable,
     Mapping,
     Optional,
+    Union,
     cast,
 )
 
@@ -40,7 +43,7 @@ class AioHTTPRequestAdapter(AsyncHTTPRequestAdapter):
 
     @property
     def query_params(self) -> QueryParams:
-        return self.request.query.copy()
+        return self.request.query.copy()  # type: ignore[attr-defined]
 
     async def get_body(self) -> str:
         return (await self.request.content.read()).decode()
@@ -71,11 +74,17 @@ class AioHTTPRequestAdapter(AsyncHTTPRequestAdapter):
 
     @property
     def content_type(self) -> Optional[str]:
-        return self.request.content_type
+        return self.headers.get("content-type")
 
 
 class GraphQLView(
-    AsyncBaseHTTPView[web.Request, web.Response, web.Response, Context, RootValue]
+    AsyncBaseHTTPView[
+        web.Request,
+        Union[web.Response, web.StreamResponse],
+        web.Response,
+        Context,
+        RootValue,
+    ]
 ):
     # Mark the view as coroutine so that AIOHTTP does not confuse it with a deprecated
     # bare handler function.
@@ -131,6 +140,8 @@ class GraphQLView(
 
         if not ws_test.ok:
             try:
+                # TODO: pass this down from run to multipart thingy
+                self.request = request
                 return await self.run(request=request)
             except HTTPException as e:
                 return web.Response(
@@ -177,3 +188,26 @@ class GraphQLView(
         sub_response.content_type = "application/json"
 
         return sub_response
+
+    async def create_multipart_response(
+        self,
+        stream: Callable[[], AsyncGenerator[str, None]],
+        sub_response: web.Response,
+    ) -> web.StreamResponse:
+        # TODO: use sub response
+        response = web.StreamResponse(
+            status=200,
+            headers={
+                "Transfer-Encoding": "chunked",
+                "Content-type": "multipart/mixed;boundary=graphql;subscriptionSpec=1.0,application/json",
+            },
+            reason="OK",
+        )
+
+        await response.prepare(self.request)
+
+        async for data in stream():
+            await response.write(data.encode())
+
+        await response.write_eof()
+        return response

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/asgi/__init__.py

@@ -5,6 +5,8 @@ from datetime import timedelta
 from typing import (
     TYPE_CHECKING,
     Any,
+    AsyncIterator,
+    Callable,
     Mapping,
     Optional,
     Sequence,
@@ -14,7 +16,12 @@ from typing import (
 
 from starlette import status
 from starlette.requests import Request
-from starlette.responses import HTMLResponse, PlainTextResponse, Response
+from starlette.responses import (
+    HTMLResponse,
+    PlainTextResponse,
+    Response,
+    StreamingResponse,
+)
 from starlette.websockets import WebSocket
 
 from strawberry.asgi.handlers import (
@@ -213,3 +220,14 @@ class GraphQL(
             response.status_code = sub_response.status_code
 
         return response
+
+    async def create_multipart_response(
+        self, stream: Callable[[], AsyncIterator[str]], sub_response: Response
+    ) -> Response:
+        return StreamingResponse(
+            stream(),
+            headers={
+                "Transfer-Encoding": "chunked",
+                "Content-type": "multipart/mixed;boundary=graphql;subscriptionSpec=1.0,application/json",
+            },
+        )

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/chalice/views.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, Dict, List, Mapping, Optional, Union, cast
 
 from chalice.app import Request, Response
 from strawberry.http.exceptions import HTTPException
@@ -92,7 +92,7 @@ class GraphQLView(
         message: str,
         error_code: str,
         http_status_code: int,
-        headers: Optional[Dict[str, str]] = None,
+        headers: Optional[Dict[str, str | List[str]]] = None,
     ) -> Response:
         """
         A wrapper for error responses
@@ -117,7 +117,7 @@ class GraphQLView(
         return Response(
             body=self.encode_json(response_data),
             status_code=status_code,
-            headers=sub_response.headers,
+            headers=sub_response.headers,  # type: ignore[arg-type]
         )
 
     def execute_request(self, request: Request) -> Response:

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/channels/handlers/http_handler.py

@@ -9,7 +9,16 @@ import json
 import warnings
 from functools import cached_property
 from io import BytesIO
-from typing import TYPE_CHECKING, Any, Dict, Mapping, Optional, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    AsyncGenerator,
+    Callable,
+    Dict,
+    Mapping,
+    Optional,
+    Union,
+)
 from urllib.parse import parse_qs
 
 from django.conf import settings
@@ -43,6 +52,14 @@ class ChannelsResponse:
     headers: Dict[bytes, bytes] = dataclasses.field(default_factory=dict)
 
 
+@dataclasses.dataclass
+class MultipartChannelsResponse:
+    stream: Callable[[], AsyncGenerator[str, None]]
+    status: int = 200
+    content_type: str = "multipart/mixed;boundary=graphql;subscriptionSpec=1.0"
+    headers: Dict[bytes, bytes] = dataclasses.field(default_factory=dict)
+
+
 @dataclasses.dataclass
 class ChannelsRequest:
     consumer: ChannelsConsumer
@@ -185,16 +202,25 @@ class BaseGraphQLHTTPConsumer(ChannelsConsumer, AsyncHttpConsumer):
     async def handle(self, body: bytes) -> None:
         request = ChannelsRequest(consumer=self, body=body)
         try:
-            response: ChannelsResponse = await self.run(request)
+            response = await self.run(request)
 
             if b"Content-Type" not in response.headers:
                 response.headers[b"Content-Type"] = response.content_type.encode()
 
-            await self.send_response(
-                response.status,
-                response.content,
-                headers=response.headers,
-            )
+            if isinstance(response, MultipartChannelsResponse):
+                response.headers[b"Transfer-Encoding"] = b"chunked"
+                await self.send_headers(headers=response.headers)
+
+                async for chunk in response.stream():
+                    # TODO: we should change more body
+                    await self.send_body(chunk.encode("utf-8"), more_body=True)
+
+            else:
+                await self.send_response(
+                    response.status,
+                    response.content,
+                    headers=response.headers,
+                )
         except HTTPException as e:
             await self.send_response(e.status_code, e.reason.encode())
 
@@ -203,7 +229,7 @@ class GraphQLHTTPConsumer(
     BaseGraphQLHTTPConsumer,
     AsyncBaseHTTPView[
         ChannelsRequest,
-        ChannelsResponse,
+        Union[ChannelsResponse, MultipartChannelsResponse],
         TemporalResponse,
         Context,
         RootValue,
@@ -247,6 +273,14 @@ class GraphQLHTTPConsumer(
     async def get_sub_response(self, request: ChannelsRequest) -> TemporalResponse:
         return TemporalResponse()
 
+    async def create_multipart_response(
+        self,
+        stream: Callable[[], AsyncGenerator[str, None]],
+        sub_response: TemporalResponse,
+    ) -> MultipartChannelsResponse:
+        # TODO: sub response
+        return MultipartChannelsResponse(stream=stream)
+
     async def render_graphql_ide(self, request: ChannelsRequest) -> ChannelsResponse:
         return ChannelsResponse(
             content=self.graphql_ide_html.encode(), content_type="text/html"
@@ -301,5 +335,5 @@ class SyncGraphQLHTTPConsumer(
         request: ChannelsRequest,
         context: Optional[Context] = UNSET,
         root_value: Optional[RootValue] = UNSET,
-    ) -> ChannelsResponse:
+    ) -> ChannelsResponse:  # pyright: ignore
         return super().run(request, context, root_value)

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/django/views.py

@@ -5,6 +5,7 @@ import warnings
 from typing import (
     TYPE_CHECKING,
     Any,
+    AsyncIterator,
     Callable,
     Mapping,
     Optional,
@@ -14,8 +15,14 @@ from typing import (
 
 from asgiref.sync import markcoroutinefunction
 from django.core.serializers.json import DjangoJSONEncoder
-from django.http import HttpRequest, HttpResponseNotAllowed, JsonResponse
-from django.http.response import HttpResponse
+from django.http import (
+    HttpRequest,
+    HttpResponse,
+    HttpResponseNotAllowed,
+    JsonResponse,
+    StreamingHttpResponse,
+)
+from django.http.response import HttpResponseBase
 from django.template import RequestContext, Template
 from django.template.exceptions import TemplateDoesNotExist
 from django.template.loader import render_to_string
@@ -116,7 +123,7 @@ class AsyncDjangoHTTPRequestAdapter(AsyncHTTPRequestAdapter):
 
     @property
     def content_type(self) -> Optional[str]:
-        return self.request.content_type
+        return self.headers.get("Content-type")
 
     async def get_body(self) -> str:
         return self.request.body.decode()
@@ -159,8 +166,9 @@ class BaseView:
 
     def create_response(
         self, response_data: GraphQLHTTPResponse, sub_response: HttpResponse
-    ) -> HttpResponse:
+    ) -> HttpResponseBase:
         data = self.encode_json(response_data)
+
         response = HttpResponse(
             data,
             content_type="application/json",
@@ -177,6 +185,17 @@ class BaseView:
 
         return response
 
+    async def create_multipart_response(
+        self, stream: Callable[[], AsyncIterator[Any]], sub_response: HttpResponse
+    ) -> HttpResponseBase:
+        return StreamingHttpResponse(
+            streaming_content=stream(),
+            headers={
+                "Transfer-Encoding": "chunked",
+                "Content-type": "multipart/mixed;boundary=graphql;subscriptionSpec=1.0,application/json",
+            },
+        )
+
     def encode_json(self, response_data: GraphQLHTTPResponse) -> str:
         return json.dumps(response_data, cls=DjangoJSONEncoder)
 
@@ -184,7 +203,7 @@ class BaseView:
 class GraphQLView(
     BaseView,
     SyncBaseHTTPView[
-        HttpRequest, HttpResponse, TemporalHttpResponse, Context, RootValue
+        HttpRequest, HttpResponseBase, TemporalHttpResponse, Context, RootValue
     ],
     View,
 ):
@@ -207,7 +226,7 @@ class GraphQLView(
     @method_decorator(csrf_exempt)
     def dispatch(
         self, request: HttpRequest, *args: Any, **kwargs: Any
-    ) -> Union[HttpResponseNotAllowed, TemplateResponse, HttpResponse]:
+    ) -> Union[HttpResponseNotAllowed, TemplateResponse, HttpResponseBase]:
         try:
             return self.run(request=request)
         except HTTPException as e:
@@ -233,7 +252,7 @@ class GraphQLView(
 class AsyncGraphQLView(
     BaseView,
     AsyncBaseHTTPView[
-        HttpRequest, HttpResponse, TemporalHttpResponse, Context, RootValue
+        HttpRequest, HttpResponseBase, TemporalHttpResponse, Context, RootValue
     ],
     View,
 ):
@@ -266,7 +285,7 @@ class AsyncGraphQLView(
     @method_decorator(csrf_exempt)
     async def dispatch(  # pyright: ignore
         self, request: HttpRequest, *args: Any, **kwargs: Any
-    ) -> Union[HttpResponseNotAllowed, TemplateResponse, HttpResponse]:
+    ) -> Union[HttpResponseNotAllowed, TemplateResponse, HttpResponseBase]:
         try:
             return await self.run(request=request)
         except HTTPException as e:

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/fastapi/router.py

@@ -6,6 +6,7 @@ from inspect import signature
 from typing import (
     TYPE_CHECKING,
     Any,
+    AsyncIterator,
     Awaitable,
     Callable,
     Mapping,
@@ -22,6 +23,7 @@ from starlette.responses import (
     HTMLResponse,
     PlainTextResponse,
     Response,
+    StreamingResponse,
 )
 from starlette.websockets import WebSocket
 
@@ -322,3 +324,14 @@ class GraphQLRouter(
         response.headers.raw.extend(sub_response.headers.raw)
 
         return response
+
+    async def create_multipart_response(
+        self, stream: Callable[[], AsyncIterator[str]], sub_response: Response
+    ) -> Response:
+        return StreamingResponse(
+            stream(),
+            headers={
+                "Transfer-Encoding": "chunked",
+                "Content-type": "multipart/mixed;boundary=graphql;subscriptionSpec=1.0,application/json",
+            },
+        )

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/flask/views.py

@@ -1,7 +1,15 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, Any, List, Mapping, Optional, Union, cast
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    List,
+    Mapping,
+    Optional,
+    Union,
+    cast,
+)
 
 from flask import Request, Response, render_template_string, request
 from flask.views import View

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/http/async_base_view.py

@@ -1,12 +1,17 @@
 import abc
+import asyncio
+import contextlib
 import json
 from typing import (
+    Any,
+    AsyncGenerator,
     Callable,
     Dict,
     Generic,
     List,
     Mapping,
     Optional,
+    Tuple,
     Union,
 )
 
@@ -15,15 +20,20 @@ from graphql import GraphQLError
 from strawberry import UNSET
 from strawberry.exceptions import MissingQueryError
 from strawberry.file_uploads.utils import replace_placeholders_with_files
-from strawberry.http import GraphQLHTTPResponse, GraphQLRequestData, process_result
+from strawberry.http import (
+    GraphQLHTTPResponse,
+    GraphQLRequestData,
+    process_result,
+)
 from strawberry.http.ides import GraphQL_IDE
 from strawberry.schema.base import BaseSchema
 from strawberry.schema.exceptions import InvalidOperationTypeError
-from strawberry.types import ExecutionResult
+from strawberry.types import ExecutionResult, SubscriptionExecutionResult
 from strawberry.types.graphql import OperationType
 
 from .base import BaseView
 from .exceptions import HTTPException
+from .parse_content_type import parse_content_type
 from .types import FormData, HTTPMethod, QueryParams
 from .typevars import Context, Request, Response, RootValue, SubResponse
 
@@ -94,9 +104,14 @@ class AsyncBaseHTTPView(
     async def render_graphql_ide(self, request: Request) -> Response:
         ...
 
+    async def create_multipart_response(
+        self, stream: Callable[[], AsyncGenerator[str, None]], sub_response: SubResponse
+    ) -> Response:
+        raise ValueError("Multipart responses are not supported")
+
     async def execute_operation(
         self, request: Request, context: Context, root_value: Optional[RootValue]
-    ) -> ExecutionResult:
+    ) -> Union[ExecutionResult, SubscriptionExecutionResult]:
         request_adapter = self.request_adapter_class(request)
 
         try:
@@ -192,6 +207,11 @@ class AsyncBaseHTTPView(
         except MissingQueryError as e:
             raise HTTPException(400, "No GraphQL query found in the request") from e
 
+        if isinstance(result, SubscriptionExecutionResult):
+            stream = self._get_stream(request, result)
+
+            return await self.create_multipart_response(stream, sub_response)
+
         response_data = await self.process_result(request=request, result=result)
 
         if result.errors:
@@ -201,15 +221,106 @@ class AsyncBaseHTTPView(
             response_data=response_data, sub_response=sub_response
         )
 
+    def encode_multipart_data(self, data: Any, separator: str) -> str:
+        return "".join(
+            [
+                f"\r\n--{separator}\r\n",
+                "Content-Type: application/json\r\n\r\n",
+                self.encode_json(data),
+                "\n",
+            ]
+        )
+
+    def _stream_with_heartbeat(
+        self, stream: Callable[[], AsyncGenerator[str, None]]
+    ) -> Callable[[], AsyncGenerator[str, None]]:
+        """Adds a heartbeat to the stream, to prevent the connection from closing
+        when there are no messages being sent."""
+        # TODO: handle errors
+        # TODO: should we do this more efficiently? and only send the heartbeat when
+        # 5 seconds have passed after the last message? (apollo router doesn't seem to do this)
+        queue = asyncio.Queue[Tuple[bool, Any]](1)
+
+        cancelling = False
+
+        async def drain():
+            try:
+                async for item in stream():
+                    await queue.put((False, item))
+            except Exception as e:
+                if not cancelling:
+                    await queue.put((True, e))
+                else:
+                    raise
+
+        async def heartbeat():
+            while True:
+                await queue.put((False, self.encode_multipart_data({}, "graphql")))
+
+                await asyncio.sleep(5)
+
+        async def merged() -> AsyncGenerator[str, None]:
+            heartbeat_task = asyncio.create_task(heartbeat())
+            task = asyncio.create_task(drain())
+
+            async def cancel_tasks():
+                nonlocal cancelling
+                cancelling = True
+                task.cancel()
+
+                with contextlib.suppress(asyncio.CancelledError):
+                    await task
+
+                heartbeat_task.cancel()
+
+                with contextlib.suppress(asyncio.CancelledError):
+                    await heartbeat_task
+
+            try:
+                while not task.done():
+                    raised, data = await queue.get()
+
+                    if raised:
+                        await cancel_tasks()
+                        raise data
+
+                    yield data
+            finally:
+                await cancel_tasks()
+
+        return merged
+
+    def _get_stream(
+        self,
+        request: Request,
+        result: SubscriptionExecutionResult,
+        separator: str = "graphql",
+    ) -> Callable[[], AsyncGenerator[str, None]]:
+        async def stream():
+            async for value in result:
+                response = await self.process_result(request, value)
+                yield self.encode_multipart_data({"payload": response}, separator)
+
+            yield f"\r\n--{separator}--\r\n"
+
+        return self._stream_with_heartbeat(stream)
+
+    async def parse_multipart_subscriptions(
+        self, request: AsyncHTTPRequestAdapter
+    ) -> Dict[str, str]:
+        return self.parse_json(await request.get_body())
+
     async def parse_http_body(
         self, request: AsyncHTTPRequestAdapter
     ) -> GraphQLRequestData:
-        content_type = request.content_type or ""
+        content_type, params = parse_content_type(request.content_type or "")
 
-        if "application/json" in content_type:
+        if content_type == "application/json":
             data = self.parse_json(await request.get_body())
-        elif content_type.startswith("multipart/form-data"):
+        elif content_type == "multipart/form-data":
             data = await self.parse_multipart(request)
+        elif self._is_multipart_subscriptions(content_type, params):
+            data = await self.parse_multipart_subscriptions(request)
         elif request.method == "GET":
             data = self.parse_query_params(request.query_params)
         else:

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/http/base.py

@@ -76,3 +76,16 @@ class BaseView(Generic[Request]):
             replace_variables=self._ide_replace_variables,
             graphql_ide=self.graphql_ide,
         )
+
+    def _is_multipart_subscriptions(
+        self, content_type: str, params: Dict[str, str]
+    ) -> bool:
+        if content_type != "multipart/mixed":
+            return False
+
+        if params.get("boundary") != "graphql":
+            return False
+
+        return tuple(
+            part.strip() for part in params.get("subscriptionspec", "").split(",")
+        ) == ("1.0", "application/json")

strawberry_graphql-0.218.0.dev1705418681/strawberry/http/parse_content_type.py

@@ -0,0 +1,17 @@
+from email.message import Message
+from typing import Dict, Tuple
+
+
+def parse_content_type(content_type: str) -> Tuple[str, Dict[str, str]]:
+    """Parse a content type header into a mime-type and a dictionary of parameters."""
+
+    email = Message()
+    email["content-type"] = content_type
+
+    params = email.get_params()
+
+    assert params
+
+    mime_type, _ = params.pop(0)
+
+    return mime_type, dict(params)

{strawberry_graphql-0.217.0 → strawberry_graphql-0.218.0.dev1705418681}/strawberry/http/sync_base_view.py

@@ -16,7 +16,11 @@ from graphql import GraphQLError
 from strawberry import UNSET
 from strawberry.exceptions import MissingQueryError
 from strawberry.file_uploads.utils import replace_placeholders_with_files
-from strawberry.http import GraphQLHTTPResponse, GraphQLRequestData, process_result
+from strawberry.http import (
+    GraphQLHTTPResponse,
+    GraphQLRequestData,
+    process_result,
+)
 from strawberry.http.ides import GraphQL_IDE
 from strawberry.schema import BaseSchema
 from strawberry.schema.exceptions import InvalidOperationTypeError
@@ -25,6 +29,7 @@ from strawberry.types.graphql import OperationType
 
 from .base import BaseView
 from .exceptions import HTTPException
+from .parse_content_type import parse_content_type
 from .types import HTTPMethod, QueryParams
 from .typevars import Context, Request, Response, RootValue, SubResponse
 
@@ -144,14 +149,18 @@ class SyncBaseHTTPView(
             raise HTTPException(400, "File(s) missing in form data") from e
 
     def parse_http_body(self, request: SyncHTTPRequestAdapter) -> GraphQLRequestData:
-        content_type = request.content_type or ""
+        content_type, params = parse_content_type(request.content_type or "")
 
-        if "application/json" in content_type:
+        if content_type == "application/json":
             data = self.parse_json(request.body)
-        elif content_type.startswith("multipart/form-data"):
+        elif content_type == "multipart/form-data":
             data = self.parse_multipart(request)
         elif request.method == "GET":
             data = self.parse_query_params(request.query_params)
+        elif self._is_multipart_subscriptions(content_type, params):
+            raise HTTPException(
+                400, "Multipart subcriptions are not supported in sync mode"
+            )
         else:
             raise HTTPException(400, "Unsupported content type")