strawberry-graphql 0.235.1.dev1719337273__py3-none-any.whl → 0.236.0__py3-none-any.whl

strawberry/fastapi/handlers/graphql_transport_ws_handler.py

@@ -15,3 +15,6 @@ class GraphQLTransportWSHandler(BaseGraphQLTransportWSHandler):
 
     async def get_root_value(self) -> Any:
         return await self._get_root_value()
+
+
+__all__ = ["GraphQLTransportWSHandler"]

strawberry/fastapi/handlers/graphql_ws_handler.py

@@ -13,3 +13,6 @@ class GraphQLWSHandler(BaseGraphQLWSHandler):
 
     async def get_root_value(self) -> Any:
         return await self._get_root_value()
+
+
+__all__ = ["GraphQLWSHandler"]

strawberry/fastapi/router.py

@@ -364,3 +364,6 @@ class GraphQLRouter(
         response.headers.raw.extend(sub_response.headers.raw)
 
         return response
+
+
+__all__ = ["GraphQLRouter"]

strawberry/federation/argument.py

@@ -1,6 +1,6 @@
 from typing import Iterable, Optional
 
-from strawberry.arguments import StrawberryArgumentAnnotation
+from strawberry.types.arguments import StrawberryArgumentAnnotation
 
 
 def argument(
@@ -27,3 +27,6 @@ def argument(
         deprecation_reason=deprecation_reason,
         directives=directives,
     )
+
+
+__all__ = ["argument"]

strawberry/federation/enum.py

@@ -11,8 +11,8 @@ from typing import (
     overload,
 )
 
-from strawberry.enum import _process_enum
-from strawberry.enum import enum_value as base_enum_value
+from strawberry.types.enum import _process_enum
+from strawberry.types.enum import enum_value as base_enum_value
 
 if TYPE_CHECKING:
     from strawberry.enum import EnumType, EnumValueDefinition
@@ -85,7 +85,6 @@ def enum(
     If name is passed, the name of the GraphQL type will be
     the value passed of name instead of the Enum class name.
     """
-
     from strawberry.federation.schema_directives import (
         Authenticated,
         Inaccessible,
@@ -118,3 +117,6 @@ def enum(
         return wrap
 
     return wrap(_cls)  # pragma: no cover
+
+
+__all__ = ["enum", "enum_value"]

strawberry/federation/field.py

@@ -15,15 +15,15 @@ from typing import (
     overload,
 )
 
-from strawberry.field import field as base_field
-from strawberry.unset import UNSET
+from strawberry.types.field import field as base_field
+from strawberry.types.unset import UNSET
 
 if TYPE_CHECKING:
     from typing_extensions import Literal
 
     from strawberry.extensions.field_extension import FieldExtension
-    from strawberry.field import _RESOLVER_TYPE, StrawberryField
     from strawberry.permission import BasePermission
+    from strawberry.types.field import _RESOLVER_TYPE, StrawberryField
 
     from .schema_directives import Override
 
@@ -203,3 +203,6 @@ def field(
         extensions=extensions,
         graphql_type=graphql_type,
     )
+
+
+__all__ = ["field"]

strawberry/federation/mutation.py

@@ -1,3 +1,5 @@
 from .field import field
 
 mutation = field
+
+__all__ = ["mutation"]

strawberry/federation/object_type.py

@@ -12,10 +12,10 @@ from typing import (
 )
 from typing_extensions import dataclass_transform
 
-from strawberry.field import StrawberryField
-from strawberry.field import field as base_field
-from strawberry.object_type import type as base_type
-from strawberry.unset import UNSET
+from strawberry.types.field import StrawberryField
+from strawberry.types.field import field as base_field
+from strawberry.types.object_type import type as base_type
+from strawberry.types.unset import UNSET
 
 from .field import field
 
@@ -369,3 +369,6 @@ def interface_object(
         is_interface=False,
         is_interface_object=True,
     )
+
+
+__all__ = ["type", "input", "interface", "interface_object"]

strawberry/federation/scalar.py

@@ -11,7 +11,7 @@ from typing import (
     overload,
 )
 
-from strawberry.custom_scalar import ScalarWrapper, _process_scalar
+from strawberry.types.scalar import ScalarWrapper, _process_scalar
 
 # in python 3.10+ NewType is a class
 if sys.version_info >= (3, 10):
@@ -79,28 +79,48 @@ def scalar(
 ) -> Any:
     """Annotates a class or type as a GraphQL custom scalar.
 
+    Args:
+        cls: The class or type to annotate
+        name: The GraphQL name of the scalar
+        description: The description of the scalar
+        specified_by_url: The URL of the specification
+        serialize: The function to serialize the scalar
+        parse_value: The function to parse the value
+        parse_literal: The function to parse the literal
+        directives: The directives to apply to the scalar
+        authenticated: Whether to add the @authenticated directive
+        inaccessible: Whether to add the @inaccessible directive
+        policy: The list of policy names to add to the @policy directive
+        requires_scopes: The list of scopes to add to the @requires directive
+        tags: The list of tags to add to the @tag directive
+
+    Returns:
+        The decorated class or type
+
     Example usages:
 
-    >>> strawberry.federation.scalar(
-    >>>     datetime.date,
-    >>>     serialize=lambda value: value.isoformat(),
-    >>>     parse_value=datetime.parse_date
-    >>> )
-
-    >>> Base64Encoded = strawberry.federation.scalar(
-    >>>     NewType("Base64Encoded", bytes),
-    >>>     serialize=base64.b64encode,
-    >>>     parse_value=base64.b64decode
-    >>> )
-
-    >>> @strawberry.federation.scalar(
-    >>>     serialize=lambda value: ",".join(value.items),
-    >>>     parse_value=lambda value: CustomList(value.split(","))
-    >>> )
-    >>> class CustomList:
-    >>>     def __init__(self, items):
-    >>>         self.items = items
+    ```python
+    strawberry.federation.scalar(
+        datetime.date,
+        serialize=lambda value: value.isoformat(),
+        parse_value=datetime.parse_date,
+    )
+
+    Base64Encoded = strawberry.federation.scalar(
+        NewType("Base64Encoded", bytes),
+        serialize=base64.b64encode,
+        parse_value=base64.b64decode,
+    )
+
 
+    @strawberry.federation.scalar(
+        serialize=lambda value: ",".join(value.items),
+        parse_value=lambda value: CustomList(value.split(",")),
+    )
+    class CustomList:
+        def __init__(self, items):
+            self.items = items
+    ```
     """
     from strawberry.federation.schema_directives import (
         Authenticated,
@@ -146,3 +166,6 @@ def scalar(
         return wrap
 
     return wrap(cls)
+
+
+__all__ = ["scalar"]

strawberry/federation/schema.py

@@ -20,17 +20,17 @@ from typing import (
 from graphql import GraphQLError
 
 from strawberry.annotation import StrawberryAnnotation
-from strawberry.custom_scalar import scalar
 from strawberry.printer import print_schema
 from strawberry.schema import Schema as BaseSchema
-from strawberry.type import (
+from strawberry.types.base import (
     StrawberryContainer,
+    StrawberryObjectDefinition,
     WithStrawberryObjectDefinition,
     get_object_definition,
 )
 from strawberry.types.info import Info
-from strawberry.types.types import StrawberryObjectDefinition
-from strawberry.union import StrawberryUnion
+from strawberry.types.scalar import scalar
+from strawberry.types.union import StrawberryUnion
 from strawberry.utils.inspect import get_func_args
 
 from .schema_directive import StrawberryFederationSchemaDirective
@@ -38,12 +38,12 @@ from .schema_directive import StrawberryFederationSchemaDirective
 if TYPE_CHECKING:
     from graphql import ExecutionContext as GraphQLExecutionContext
 
-    from strawberry.custom_scalar import ScalarDefinition, ScalarWrapper
-    from strawberry.enum import EnumDefinition
     from strawberry.extensions import SchemaExtension
     from strawberry.federation.schema_directives import ComposeDirective
     from strawberry.schema.config import StrawberryConfig
     from strawberry.schema_directive import StrawberrySchemaDirective
+    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.scalar import ScalarDefinition, ScalarWrapper
 
 
 FederationAny = scalar(NewType("_Any", object), name="_Any")  # type: ignore
@@ -112,7 +112,6 @@ class Schema(BaseSchema):
         The _service field is added by default, but the _entities field is only
         added if the schema contains an entity type.
         """
-
         import strawberry
         from strawberry.tools.create_type import create_type
         from strawberry.tools.merge_types import merge_types
@@ -181,7 +180,7 @@ class Schema(BaseSchema):
 
                 get_result = partial(resolve_reference, **kwargs)
             else:
-                from strawberry.arguments import convert_argument
+                from strawberry.types.arguments import convert_argument
 
                 config = info.schema.config
                 scalar_registry = info.schema.schema_converter.scalar_registry
@@ -209,7 +208,8 @@ class Schema(BaseSchema):
     def _remove_resolvable_field(self) -> None:
         # this might be removed when we remove support for federation 1
         # or when we improve how we print the directives
-        from ..unset import UNSET
+        from strawberry.types.unset import UNSET
+
         from .schema_directives import Key
 
         for directive in self.schema_directives_in_use:
@@ -392,3 +392,6 @@ def _has_federation_keys(
         return any(_is_key(directive) for directive in definition.directives or [])
 
     return False
+
+
+__all__ = ["Schema"]

strawberry/federation/schema_directive.py

@@ -3,9 +3,9 @@ from typing import Callable, List, Optional, Type, TypeVar
 from typing_extensions import dataclass_transform
 
 from strawberry.directive import directive_field
-from strawberry.field import StrawberryField, field
-from strawberry.object_type import _wrap_dataclass
 from strawberry.schema_directive import Location, StrawberrySchemaDirective
+from strawberry.types.field import StrawberryField, field
+from strawberry.types.object_type import _wrap_dataclass
 from strawberry.types.type_resolver import _get_fields
 
 

strawberry/federation/schema_directives.py

@@ -3,7 +3,7 @@ from typing import ClassVar, List, Optional
 
 from strawberry import directive_field
 from strawberry.schema_directive import Location, schema_directive
-from strawberry.unset import UNSET
+from strawberry.types.unset import UNSET
 
 from .types import (
     FieldSet,
@@ -226,3 +226,21 @@ class Policy(FederationDirective):
     imported_from: ClassVar[ImportedFrom] = ImportedFrom(
         name="policy", url="https://specs.apollo.dev/federation/v2.7"
     )
+
+
+__all__ = [
+    "External",
+    "Requires",
+    "Provides",
+    "Key",
+    "Shareable",
+    "Link",
+    "Tag",
+    "Override",
+    "Inaccessible",
+    "ComposeDirective",
+    "InterfaceObject",
+    "Authenticated",
+    "RequiresScopes",
+    "Policy",
+]

strawberry/federation/types.py

@@ -1,7 +1,7 @@
 from enum import Enum
 
-from strawberry.custom_scalar import scalar
-from strawberry.enum import enum
+from strawberry.types.enum import enum
+from strawberry.types.scalar import scalar
 
 FieldSet = scalar(str, name="_FieldSet")
 
@@ -12,3 +12,6 @@ LinkImport = scalar(object, name="link__Import")
 class LinkPurpose(Enum):
     SECURITY = "SECURITY"
     EXECUTION = "EXECUTION"
+
+
+__all__ = ["FieldSet", "LinkImport", "LinkPurpose"]

strawberry/federation/union.py

@@ -1,7 +1,7 @@
 from typing import Any, Collection, Iterable, Optional, Type
 
-from strawberry.union import StrawberryUnion
-from strawberry.union import union as base_union
+from strawberry.types.union import StrawberryUnion
+from strawberry.types.union import union as base_union
 
 
 def union(
@@ -15,15 +15,31 @@ def union(
 ) -> StrawberryUnion:
     """Creates a new named Union type.
 
+    Args:
+        name: The GraphQL name of the Union type.
+        types: The types that the Union can be.
+            (Deprecated, use `Annotated[U, strawberry.federation.union("Name")]` instead)
+        description: The  GraphQL description of the Union type.
+        directives: The directives to attach to the Union type.
+        inaccessible: Whether the Union type is inaccessible.
+        tags: The federation tags to attach to the Union type.
+
     Example usages:
 
-    >>> @strawberry.type
-    ... class A: ...
-    >>> @strawberry.type
-    ... class B: ...
-    >>> strawberry.federation.union("Name", (A, Optional[B]))
-    """
+    ```python
+    import strawberry
+    from typing import Annotated
+
+    @strawberry.federation.type(keys=["id"])
+    class A:
+        id: strawberry.ID
+
+    @strawberry.federation.type(keys=["id"])
+    class B:
+        id: strawberry.ID
 
+    MyUnion = Annotated[A | B, strawberry.federation.union("Name", tags=["tag"])]
+    """
     from strawberry.federation.schema_directives import Inaccessible, Tag
 
     directives = list(directives)
@@ -40,3 +56,6 @@ def union(
         description=description,
         directives=directives,
     )
+
+
+__all__ = ["union"]

strawberry/field_extensions/input_mutation.py

@@ -8,13 +8,13 @@ from typing import (
 
 import strawberry
 from strawberry.annotation import StrawberryAnnotation
-from strawberry.arguments import StrawberryArgument
 from strawberry.extensions.field_extension import (
     AsyncExtensionResolver,
     FieldExtension,
     SyncExtensionResolver,
 )
-from strawberry.field import StrawberryField
+from strawberry.types.arguments import StrawberryArgument
+from strawberry.types.field import StrawberryField
 from strawberry.utils.str_converters import capitalize_first, to_camel_case
 
 if TYPE_CHECKING:
@@ -90,3 +90,6 @@ class InputMutationExtension(FieldExtension):
             **kwargs,
             **vars(input_args),
         )
+
+
+__all__ = ["InputMutationExtension"]

strawberry/file_uploads/scalars.py

@@ -1,5 +1,7 @@
 from typing import NewType
 
-from ..custom_scalar import scalar
+from strawberry.types.scalar import scalar
 
 Upload = scalar(NewType("Upload", bytes), parse_value=lambda x: x)
+
+__all__ = ["Upload"]

strawberry/file_uploads/utils.py

@@ -30,3 +30,6 @@ def replace_placeholders_with_files(
                 target_object[value_key] = file_object
 
     return operations
+
+
+__all__ = ["replace_placeholders_with_files"]

strawberry/flask/views.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import warnings
-from typing import TYPE_CHECKING, Any, List, Mapping, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, Mapping, Optional, Union, cast
 
 from flask import Request, Response, render_template_string, request
 from flask.views import View
@@ -26,7 +26,7 @@ class FlaskHTTPRequestAdapter(SyncHTTPRequestAdapter):
         self.request = request
 
     @property
-    def query_params(self) -> Mapping[str, Union[str, Optional[List[str]]]]:
+    def query_params(self) -> QueryParams:
         return self.request.args.to_dict()
 
     @property
@@ -177,3 +177,9 @@ class AsyncGraphQLView(
 
     async def render_graphql_ide(self, request: Request) -> Response:
         return render_template_string(self.graphql_ide_html)  # type: ignore
+
+
+__all__ = [
+    "GraphQLView",
+    "AsyncGraphQLView",
+]

strawberry/http/__init__.py

@@ -48,3 +48,12 @@ def parse_request_data(data: Mapping[str, Any]) -> GraphQLRequestData:
         variables=data.get("variables"),
         operation_name=data.get("operationName"),
     )
+
+
+__all__ = [
+    "GraphQLHTTPResponse",
+    "process_result",
+    "GraphQLRequestData",
+    "parse_query_params",
+    "parse_request_data",
+]

strawberry/http/async_base_view.py

@@ -136,9 +136,7 @@ class AsyncBaseHTTPView(
     def _handle_errors(
         self, errors: List[GraphQLError], response_data: GraphQLHTTPResponse
     ) -> None:
-        """
-        Hook to allow custom handling of errors, used by the Sentry Integration
-        """
+        """Hook to allow custom handling of errors, used by the Sentry Integration."""
 
     async def run(
         self,
@@ -213,3 +211,6 @@ class AsyncBaseHTTPView(
         self, request: Request, result: ExecutionResult
     ) -> GraphQLHTTPResponse:
         return process_result(result)
+
+
+__all__ = ["AsyncBaseHTTPView"]

strawberry/http/base.py

@@ -4,7 +4,7 @@ from typing_extensions import Protocol
 
 from strawberry.http import GraphQLHTTPResponse
 from strawberry.http.ides import GraphQL_IDE, get_graphql_ide_html
-from strawberry.http.types import HTTPMethod
+from strawberry.http.types import HTTPMethod, QueryParams
 
 from .exceptions import HTTPException
 from .typevars import Request
@@ -50,17 +50,12 @@ class BaseView(Generic[Request]):
     def encode_json(self, response_data: GraphQLHTTPResponse) -> str:
         return json.dumps(response_data)
 
-    def parse_query_params(
-        self, params: Mapping[str, Optional[Union[str, List[str]]]]
-    ) -> Dict[str, Any]:
+    def parse_query_params(self, params: QueryParams) -> Dict[str, Any]:
         params = dict(params)
 
         if "variables" in params:
             variables = params["variables"]
 
-            if isinstance(variables, list):
-                variables = variables[0]
-
             if variables:
                 params["variables"] = self.parse_json(variables)
 
@@ -73,3 +68,6 @@ class BaseView(Generic[Request]):
             replace_variables=self._ide_replace_variables,
             graphql_ide=self.graphql_ide,
         )
+
+
+__all__ = ["BaseView"]

strawberry/http/exceptions.py

@@ -2,3 +2,6 @@ class HTTPException(Exception):
     def __init__(self, status_code: int, reason: str) -> None:
         self.status_code = status_code
         self.reason = reason
+
+
+__all__ = ["HTTPException"]

strawberry/http/ides.py

@@ -28,3 +28,6 @@ def get_graphql_ide_html(
         )
 
     return template
+
+
+__all__ = ["get_graphql_ide_html", "GraphQL_IDE"]

strawberry/http/sync_base_view.py

@@ -151,9 +151,7 @@ class SyncBaseHTTPView(
     def _handle_errors(
         self, errors: List[GraphQLError], response_data: GraphQLHTTPResponse
     ) -> None:
-        """
-        Hook to allow custom handling of errors, used by the Sentry Integration
-        """
+        """Hook to allow custom handling of errors, used by the Sentry Integration."""
 
     def run(
         self,
@@ -208,3 +206,6 @@ class SyncBaseHTTPView(
         self, request: Request, result: ExecutionResult
     ) -> GraphQLHTTPResponse:
         return process_result(result)
+
+
+__all__ = ["SyncBaseHTTPView"]

strawberry/http/temporal_response.py

@@ -6,3 +6,6 @@ from typing import Dict
 class TemporalResponse:
     status_code: int = 200
     headers: Dict[str, str] = field(default_factory=dict)
+
+
+__all__ = ["TemporalResponse"]

strawberry/http/types.py

@@ -1,13 +1,16 @@
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, Mapping, Optional
 from typing_extensions import Literal, TypedDict
 
 HTTPMethod = Literal[
     "GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS", "TRACE"
 ]
 
-QueryParams = Mapping[str, Optional[Union[str, List[str]]]]
+QueryParams = Mapping[str, Optional[str]]
 
 
 class FormData(TypedDict):
     files: Mapping[str, Any]
     form: Mapping[str, Any]
+
+
+__all__ = ["HTTPMethod", "QueryParams", "FormData"]

strawberry/http/typevars.py

@@ -5,3 +5,6 @@ Response = TypeVar("Response")
 SubResponse = TypeVar("SubResponse")
 Context = TypeVar("Context")
 RootValue = TypeVar("RootValue")
+
+
+__all__ = ["Request", "Response", "SubResponse", "Context", "RootValue"]

strawberry/litestar/controller.py

@@ -433,3 +433,9 @@ def make_graphql_controller(
     _GraphQLController.graphql_ide = graphql_ide_
 
     return _GraphQLController
+
+
+__all__ = [
+    "make_graphql_controller",
+    "GraphQLController",
+]

strawberry/litestar/handlers/graphql_transport_ws_handler.py

@@ -55,3 +55,6 @@ class GraphQLTransportWSHandler(BaseGraphQLTransportWSHandler):
             pass
         finally:
             await self.shutdown()
+
+
+__all__ = ["GraphQLTransportWSHandler"]

strawberry/litestar/handlers/graphql_ws_handler.py

@@ -60,3 +60,6 @@ class GraphQLWSHandler(BaseGraphQLWSHandler):
 
             for operation_id in list(self.subscriptions.keys()):
                 await self.cleanup_operation(operation_id)
+
+
+__all__ = ["GraphQLWSHandler"]