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

strawberry/schema/schema_converter.py

@@ -44,9 +44,6 @@ from graphql import (
 from graphql.language.directive_locations import DirectiveLocation
 
 from strawberry.annotation import StrawberryAnnotation
-from strawberry.arguments import StrawberryArgument, convert_arguments
-from strawberry.custom_scalar import ScalarWrapper
-from strawberry.enum import EnumDefinition
 from strawberry.exceptions import (
     DuplicatedTypeName,
     InvalidTypeInputForUnion,
@@ -55,21 +52,24 @@ from strawberry.exceptions import (
     ScalarAlreadyRegisteredError,
     UnresolvedFieldTypeError,
 )
-from strawberry.field import UNRESOLVED
-from strawberry.lazy_type import LazyType
-from strawberry.private import is_private
 from strawberry.schema.types.scalar import _make_scalar_type
-from strawberry.type import (
+from strawberry.types.arguments import StrawberryArgument, convert_arguments
+from strawberry.types.base import (
     StrawberryList,
+    StrawberryObjectDefinition,
     StrawberryOptional,
     StrawberryType,
     get_object_definition,
     has_object_definition,
 )
+from strawberry.types.enum import EnumDefinition
+from strawberry.types.field import UNRESOLVED
 from strawberry.types.info import Info
-from strawberry.types.types import StrawberryObjectDefinition
-from strawberry.union import StrawberryUnion
-from strawberry.unset import UNSET
+from strawberry.types.lazy_type import LazyType
+from strawberry.types.private import is_private
+from strawberry.types.scalar import ScalarWrapper
+from strawberry.types.union import StrawberryUnion
+from strawberry.types.unset import UNSET
 from strawberry.utils.await_maybe import await_maybe
 
 from ..extensions.field_extension import build_field_extension_resolvers
@@ -85,12 +85,12 @@ if TYPE_CHECKING:
         GraphQLScalarType,
     )
 
-    from strawberry.custom_scalar import ScalarDefinition
     from strawberry.directive import StrawberryDirective
-    from strawberry.enum import EnumValue
-    from strawberry.field import StrawberryField
     from strawberry.schema.config import StrawberryConfig
     from strawberry.schema_directive import StrawberrySchemaDirective
+    from strawberry.types.enum import EnumValue
+    from strawberry.types.field import StrawberryField
+    from strawberry.types.scalar import ScalarDefinition
 
 
 FieldType = TypeVar(
@@ -681,7 +681,6 @@ class GraphQLCoreConverter:
 
         def wrap_field_extensions() -> Callable[..., Any]:
             """Wrap the provided field resolver with the middleware."""
-
             for extension in field.extensions:
                 extension.apply(field)
 
@@ -999,3 +998,6 @@ class GraphQLCoreConverter:
             second_origin = None
 
         raise DuplicatedTypeName(first_origin, second_origin, name)
+
+
+__all__ = ["GraphQLCoreConverter"]

strawberry/schema/types/base_scalars.py

@@ -7,7 +7,7 @@ from typing import Callable
 import dateutil.parser
 from graphql import GraphQLError
 
-from strawberry.custom_scalar import scalar
+from strawberry.types.scalar import scalar
 
 
 def wrap_parser(parser: Callable, type_: str) -> Callable:
@@ -80,3 +80,5 @@ Void = scalar(
     parse_value=_verify_void,
     description="Represents NULL values",
 )
+
+__all__ = ["Date", "DateTime", "Time", "Decimal", "UUID", "Void"]

strawberry/schema/types/concrete_type.py

@@ -6,10 +6,10 @@ from typing import TYPE_CHECKING, Dict, Union
 from graphql import GraphQLField, GraphQLInputField, GraphQLType
 
 if TYPE_CHECKING:
-    from strawberry.custom_scalar import ScalarDefinition
-    from strawberry.enum import EnumDefinition
-    from strawberry.types.types import StrawberryObjectDefinition
-    from strawberry.union import StrawberryUnion
+    from strawberry.types.base import StrawberryObjectDefinition
+    from strawberry.types.enum import EnumDefinition
+    from strawberry.types.scalar import ScalarDefinition
+    from strawberry.types.union import StrawberryUnion
 
 Field = Union[GraphQLInputField, GraphQLField]
 

strawberry/schema/types/scalar.py

@@ -12,11 +12,11 @@ from graphql import (
     GraphQLString,
 )
 
-from strawberry.custom_scalar import ScalarDefinition, scalar
 from strawberry.file_uploads.scalars import Upload
 from strawberry.relay.types import GlobalID
 from strawberry.scalars import ID
 from strawberry.schema.types import base_scalars
+from strawberry.types.scalar import ScalarDefinition, scalar
 
 
 def _make_scalar_type(definition: ScalarDefinition) -> GraphQLScalarType:
@@ -78,3 +78,10 @@ DEFAULT_SCALAR_REGISTRY: Dict[object, ScalarDefinition] = {
         )
     ),
 }
+
+__all__ = [
+    "DEFAULT_SCALAR_REGISTRY",
+    "_get_scalar_definition",
+    "_make_scalar_definition",
+    "_make_scalar_type",
+]

strawberry/schema/validation_rules/one_of.py

@@ -78,3 +78,6 @@ class OneOfInputValidationRule(ValidationRule):
                         nodes=[node],
                     )
                 )
+
+
+__all__ = ["OneOfInputValidationRule"]

strawberry/schema_codegen/__init__.py

@@ -802,3 +802,6 @@ def codegen(schema: str) -> str:
     module = cst.Module(body=body)  # type: ignore
 
     return module.code
+
+
+__all__ = ["codegen"]

strawberry/schema_directive.py

@@ -3,11 +3,11 @@ from enum import Enum
 from typing import Callable, List, Optional, Type, TypeVar
 from typing_extensions import dataclass_transform
 
-from strawberry.object_type import _wrap_dataclass
+from strawberry.types.field import StrawberryField, field
+from strawberry.types.object_type import _wrap_dataclass
 from strawberry.types.type_resolver import _get_fields
 
 from .directive import directive_field
-from .field import StrawberryField, field
 
 
 class Location(Enum):

strawberry/starlite/controller.py

@@ -382,3 +382,6 @@ def make_graphql_controller(
             )
 
     return GraphQLController
+
+
+__all__ = ["make_graphql_controller"]

strawberry/starlite/handlers/graphql_transport_ws_handler.py

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

strawberry/starlite/handlers/graphql_ws_handler.py

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

strawberry/subscriptions/__init__.py

@@ -1,2 +1,8 @@
 GRAPHQL_TRANSPORT_WS_PROTOCOL = "graphql-transport-ws"
 GRAPHQL_WS_PROTOCOL = "graphql-ws"
+
+
+__all__ = [
+    "GRAPHQL_TRANSPORT_WS_PROTOCOL",
+    "GRAPHQL_WS_PROTOCOL",
+]

strawberry/subscriptions/protocols/graphql_transport_ws/__init__.py

@@ -1,2 +1,7 @@
 # Code 4406 is "Subprotocol not acceptable"
 WS_4406_PROTOCOL_NOT_ACCEPTABLE = 4406
+
+
+__all__ = [
+    "WS_4406_PROTOCOL_NOT_ACCEPTABLE",
+]

strawberry/subscriptions/protocols/graphql_transport_ws/handlers.py

@@ -30,7 +30,7 @@ from strawberry.subscriptions.protocols.graphql_transport_ws.types import (
     SubscribeMessagePayload,
 )
 from strawberry.types.graphql import OperationType
-from strawberry.unset import UNSET
+from strawberry.types.unset import UNSET
 from strawberry.utils.debug import pretty_print_graphql_operation
 from strawberry.utils.operation import get_operation_type
 
@@ -66,23 +66,23 @@ class BaseGraphQLTransportWSHandler(ABC):
 
     @abstractmethod
     async def get_context(self) -> Any:
-        """Return the operations context"""
+        """Return the operations context."""
 
     @abstractmethod
     async def get_root_value(self) -> Any:
-        """Return the schemas root value"""
+        """Return the schemas root value."""
 
     @abstractmethod
     async def send_json(self, data: dict) -> None:
-        """Send the data JSON encoded to the WebSocket client"""
+        """Send the data JSON encoded to the WebSocket client."""
 
     @abstractmethod
     async def close(self, code: int, reason: str) -> None:
-        """Close the WebSocket with the passed code and reason"""
+        """Close the WebSocket with the passed code and reason."""
 
     @abstractmethod
     async def handle_request(self) -> Any:
-        """Handle the request this instance was created for"""
+        """Handle the request this instance was created for."""
 
     async def handle(self) -> Any:
         return await self.handle_request()
@@ -286,10 +286,7 @@ class BaseGraphQLTransportWSHandler(ABC):
     async def operation_task(
         self, result_source: AsyncGenerator, operation: Operation
     ) -> None:
-        """
-        Operation task top level method.  Cleans up and de-registers the operation
-        once it is done.
-        """
+        """The operation task's top level method. Cleans-up and de-registers the operation once it is done."""
         # TODO: Handle errors in this method using self.handle_task_exception()
         try:
             await self.handle_async_results(result_source, operation)
@@ -371,9 +368,7 @@ class BaseGraphQLTransportWSHandler(ABC):
         # websocket handler Task.
 
     async def reap_completed_tasks(self) -> None:
-        """
-        Await tasks that have completed
-        """
+        """Await tasks that have completed."""
         tasks, self.completed_tasks = self.completed_tasks, []
         for task in tasks:
             with suppress(BaseException):
@@ -381,10 +376,7 @@ class BaseGraphQLTransportWSHandler(ABC):
 
 
 class Operation:
-    """
-    A class encapsulating a single operation with its id.
-    Helps enforce protocol state transition.
-    """
+    """A class encapsulating a single operation with its id. Helps enforce protocol state transition."""
 
     __slots__ = ["handler", "id", "operation_type", "completed", "task"]
 
@@ -408,3 +400,6 @@ class Operation:
             # de-register the operation _before_ sending the final message
             self.handler.forget_id(self.id)
         await self.handler.send_message(message)
+
+
+__all__ = ["BaseGraphQLTransportWSHandler", "Operation"]

strawberry/subscriptions/protocols/graphql_transport_ws/types.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import asdict, dataclass
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-from strawberry.unset import UNSET
+from strawberry.types.unset import UNSET
 
 if TYPE_CHECKING:
     from graphql import GraphQLFormattedError
@@ -21,9 +21,7 @@ class GraphQLTransportMessage:
 
 @dataclass
 class ConnectionInitMessage(GraphQLTransportMessage):
-    """
-    Direction: Client -> Server
-    """
+    """Direction: Client -> Server."""
 
     payload: Optional[Dict[str, Any]] = UNSET
     type: str = "connection_init"
@@ -31,9 +29,7 @@ class ConnectionInitMessage(GraphQLTransportMessage):
 
 @dataclass
 class ConnectionAckMessage(GraphQLTransportMessage):
-    """
-    Direction: Server -> Client
-    """
+    """Direction: Server -> Client."""
 
     payload: Optional[Dict[str, Any]] = UNSET
     type: str = "connection_ack"
@@ -41,9 +37,7 @@ class ConnectionAckMessage(GraphQLTransportMessage):
 
 @dataclass
 class PingMessage(GraphQLTransportMessage):
-    """
-    Direction: bidirectional
-    """
+    """Direction: bidirectional."""
 
     payload: Optional[Dict[str, Any]] = UNSET
     type: str = "ping"
@@ -51,9 +45,7 @@ class PingMessage(GraphQLTransportMessage):
 
 @dataclass
 class PongMessage(GraphQLTransportMessage):
-    """
-    Direction: bidirectional
-    """
+    """Direction: bidirectional."""
 
     payload: Optional[Dict[str, Any]] = UNSET
     type: str = "pong"
@@ -69,9 +61,7 @@ class SubscribeMessagePayload:
 
 @dataclass
 class SubscribeMessage(GraphQLTransportMessage):
-    """
-    Direction: Client -> Server
-    """
+    """Direction: Client -> Server."""
 
     id: str
     payload: SubscribeMessagePayload
@@ -80,9 +70,7 @@ class SubscribeMessage(GraphQLTransportMessage):
 
 @dataclass
 class NextMessage(GraphQLTransportMessage):
-    """
-    Direction: Server -> Client
-    """
+    """Direction: Server -> Client."""
 
     id: str
     payload: Dict[str, Any]  # TODO: shape like FormattedExecutionResult
@@ -94,9 +82,7 @@ class NextMessage(GraphQLTransportMessage):
 
 @dataclass
 class ErrorMessage(GraphQLTransportMessage):
-    """
-    Direction: Server -> Client
-    """
+    """Direction: Server -> Client."""
 
     id: str
     payload: List[GraphQLFormattedError]
@@ -105,9 +91,19 @@ class ErrorMessage(GraphQLTransportMessage):
 
 @dataclass
 class CompleteMessage(GraphQLTransportMessage):
-    """
-    Direction: bidirectional
-    """
+    """Direction: bidirectional."""
 
     id: str
     type: str = "complete"
+
+
+__all__ = [
+    "ConnectionInitMessage",
+    "ConnectionAckMessage",
+    "PingMessage",
+    "PongMessage",
+    "SubscribeMessage",
+    "NextMessage",
+    "ErrorMessage",
+    "CompleteMessage",
+]

strawberry/subscriptions/protocols/graphql_ws/__init__.py

@@ -8,3 +8,17 @@ GQL_DATA = "data"
 GQL_ERROR = "error"
 GQL_COMPLETE = "complete"
 GQL_STOP = "stop"
+
+
+__all__ = [
+    "GQL_CONNECTION_INIT",
+    "GQL_CONNECTION_ACK",
+    "GQL_CONNECTION_ERROR",
+    "GQL_CONNECTION_TERMINATE",
+    "GQL_CONNECTION_KEEP_ALIVE",
+    "GQL_START",
+    "GQL_DATA",
+    "GQL_ERROR",
+    "GQL_COMPLETE",
+    "GQL_STOP",
+]

strawberry/subscriptions/protocols/graphql_ws/handlers.py

@@ -51,23 +51,23 @@ class BaseGraphQLWSHandler(ABC):
 
     @abstractmethod
     async def get_context(self) -> Any:
-        """Return the operations context"""
+        """Return the operations context."""
 
     @abstractmethod
     async def get_root_value(self) -> Any:
-        """Return the schemas root value"""
+        """Return the schemas root value."""
 
     @abstractmethod
     async def send_json(self, data: OperationMessage) -> None:
-        """Send the data JSON encoded to the WebSocket client"""
+        """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"""
+        """Close the WebSocket with the passed code and reason."""
 
     @abstractmethod
     async def handle_request(self) -> Any:
-        """Handle the request this instance was created for"""
+        """Handle the request this instance was created for."""
 
     async def handle(self) -> Any:
         return await self.handle_request()
@@ -208,3 +208,6 @@ class BaseGraphQLWSHandler(ABC):
         if payload is not None:
             data["payload"] = payload
         await self.send_json(data)
+
+
+__all__ = ["BaseGraphQLWSHandler"]

strawberry/subscriptions/protocols/graphql_ws/types.py

@@ -38,3 +38,14 @@ class OperationMessage(TypedDict, total=False):
     type: str
     id: str
     payload: OperationMessagePayload
+
+
+__all__ = [
+    "ConnectionInitPayload",
+    "ConnectionErrorPayload",
+    "StartPayload",
+    "DataPayload",
+    "ErrorPayload",
+    "OperationMessagePayload",
+    "OperationMessage",
+]

strawberry/test/client.py

@@ -91,47 +91,59 @@ class BaseGraphQLTestClient(ABC):
     def _build_multipart_file_map(
         variables: Dict[str, Mapping], files: Dict[str, object]
     ) -> Dict[str, List[str]]:
-        """Creates the file mapping between the variables and the files objects passed
-        as key arguments
+        """Creates the file mapping between the variables and the files objects passed as key arguments.
+
+        Args:
+            variables: A dictionary with the variables that are going to be passed to the
+                query.
+            files: A dictionary with the files that are going to be passed to the query.
 
         Example usages:
 
-        >>> _build_multipart_file_map(
-        >>>     variables={"textFile": None}, files={"textFile": f}
-        >>> )
-        ... {"textFile": ["variables.textFile"]}
+        ```python
+        _build_multipart_file_map(variables={"textFile": None}, files={"textFile": f})
+        # {"textFile": ["variables.textFile"]}
+        ```
 
         If the variable is a list we have to enumerate files in the mapping
-        >>> _build_multipart_file_map(
-        >>>     variables={"files": [None, None]},
-        >>>     files={"file1": file1, "file2": file2},
-        >>> )
-        ... {"file1": ["variables.files.0"], "file2": ["variables.files.1"]}
+
+        ```python
+        _build_multipart_file_map(
+            variables={"files": [None, None]},
+            files={"file1": file1, "file2": file2},
+        )
+        # {"file1": ["variables.files.0"], "file2": ["variables.files.1"]}
+        ```
 
         If `variables` contains another keyword (a folder) we must include that keyword
         in the mapping
-        >>> _build_multipart_file_map(
-        >>>     variables={"folder": {"files": [None, None]}},
-        >>>     files={"file1": file1, "file2": file2},
-        >>> )
-        ... {
-        ...     "file1": ["variables.files.folder.files.0"],
-        ...     "file2": ["variables.files.folder.files.1"]
-        ... }
+
+        ```python
+        _build_multipart_file_map(
+            variables={"folder": {"files": [None, None]}},
+            files={"file1": file1, "file2": file2},
+        )
+        # {
+        #     "file1": ["variables.files.folder.files.0"],
+        #     "file2": ["variables.files.folder.files.1"]
+        # }
+        ```
 
         If `variables` includes both a list of files and other single values, we must
         map them accordingly
-        >>> _build_multipart_file_map(
-        >>>     variables={"files": [None, None], "textFile": None},
-        >>>     files={"file1": file1, "file2": file2, "textFile": file3},
-        >>> )
-        ... {
-        ...     "file1": ["variables.files.0"],
-        ...     "file2": ["variables.files.1"],
-        ...     "textFile": ["variables.textFile"],
-        ... }
-        """
 
+        ```python
+        _build_multipart_file_map(
+            variables={"files": [None, None], "textFile": None},
+            files={"file1": file1, "file2": file2, "textFile": file3},
+        )
+        # {
+        #     "file1": ["variables.files.0"],
+        #     "file2": ["variables.files.1"],
+        #     "textFile": ["variables.textFile"],
+        # }
+        ```
+        """
         map: Dict[str, List[str]] = {}
         for key, values in variables.items():
             reference = key
@@ -167,3 +179,6 @@ class BaseGraphQLTestClient(ABC):
         if type == "multipart":
             return json.loads(response.content.decode())
         return response.json()
+
+
+__all__ = ["BaseGraphQLTestClient", "Response", "Body"]

strawberry/tools/create_type.py

@@ -2,7 +2,7 @@ import types
 from typing import List, Optional, Sequence, Type
 
 import strawberry
-from strawberry.field import StrawberryField
+from strawberry.types.field import StrawberryField
 
 
 def create_type(
@@ -14,15 +14,31 @@ def create_type(
     directives: Optional[Sequence[object]] = (),
     extend: bool = False,
 ) -> Type:
-    """Create a Strawberry type from a list of StrawberryFields
+    """Create a Strawberry type from a list of StrawberryFields.
 
-    >>> @strawberry.field
-    >>> def hello(info) -> str:
-    >>>     return "World"
-    >>>
-    >>> Query = create_type(name="Query", fields=[hello])
-    """
+    Args:
+        name: The GraphQL name of the type.
+        fields: The fields of the type.
+        is_input: Whether the type is an input type.
+        is_interface: Whether the type is an interface.
+        description: The GraphQL description of the type.
+        directives: The directives to attach to the type.
+        extend: Whether the type is an extension.
+
+    Example usage:
+
+    ```python
+    import strawberry
+
+
+    @strawberry.field
+    def hello(info) -> str:
+        return "World"
 
+
+    Query = create_type(name="Query", fields=[hello])
+    ```
+    """
     if not fields:
         raise ValueError(f'Can\'t create type "{name}" with no fields')
 
@@ -55,3 +71,6 @@ def create_type(
         directives=directives,
         extend=extend,
     )
+
+
+__all__ = ["create_type"]

strawberry/tools/merge_types.py

@@ -4,11 +4,11 @@ from itertools import chain
 from typing import Tuple
 
 import strawberry
-from strawberry.type import has_object_definition
+from strawberry.types.base import has_object_definition
 
 
 def merge_types(name: str, types: Tuple[type, ...]) -> type:
-    """Merge multiple Strawberry types into one
+    """Merge multiple Strawberry types into one.
 
     For example, given two queries `A` and `B`, one can merge them into a
     super type as follows:
@@ -20,7 +20,6 @@ def merge_types(name: str, types: Tuple[type, ...]) -> type:
         class SuperQuery(B, A):
             ...
     """
-
     if not types:
         raise ValueError("Can't merge types if none are supplied")
 
@@ -35,3 +34,6 @@ def merge_types(name: str, types: Tuple[type, ...]) -> type:
         )
 
     return strawberry.type(type(name, types, {}))
+
+
+__all__ = ["merge_types"]

strawberry/types/__init__.py

@@ -1,4 +1,11 @@
+from .base import get_object_definition, has_object_definition
 from .execution import ExecutionContext, ExecutionResult
 from .info import Info
 
-__all__ = ["ExecutionContext", "ExecutionResult", "Info"]
+__all__ = [
+    "ExecutionContext",
+    "ExecutionResult",
+    "Info",
+    "get_object_definition",
+    "has_object_definition",
+]