strawberry-graphql 0.254.1__py3-none-any.whl → 0.256.0__py3-none-any.whl

strawberry/codegen/plugins/print_operation.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import textwrap
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, Optional
 
 from strawberry.codegen import CodegenFile, QueryCodegenPlugin
 from strawberry.codegen.types import (
@@ -35,8 +35,8 @@ if TYPE_CHECKING:
 
 class PrintOperationPlugin(QueryCodegenPlugin):
     def generate_code(
-        self, types: List[GraphQLType], operation: GraphQLOperation
-    ) -> List[CodegenFile]:
+        self, types: list[GraphQLType], operation: GraphQLOperation
+    ) -> list[CodegenFile]:
         code_lines = []
         for t in types:
             if not isinstance(t, GraphQLFragmentType):
@@ -139,7 +139,7 @@ class PrintOperationPlugin(QueryCodegenPlugin):
 
         raise ValueError(f"not supported: {type(value)}")  # pragma: no cover
 
-    def _print_arguments(self, arguments: List[GraphQLArgument]) -> str:
+    def _print_arguments(self, arguments: list[GraphQLArgument]) -> str:
         if not arguments:
             return ""
 
@@ -154,7 +154,7 @@ class PrintOperationPlugin(QueryCodegenPlugin):
             + ")"
         )
 
-    def _print_directives(self, directives: List[GraphQLDirective]) -> str:
+    def _print_directives(self, directives: list[GraphQLDirective]) -> str:
         if not directives:
             return ""
 
@@ -204,7 +204,7 @@ class PrintOperationPlugin(QueryCodegenPlugin):
 
         raise ValueError(f"Unsupported selection: {selection}")  # pragma: no cover
 
-    def _print_selections(self, selections: List[GraphQLSelection]) -> str:
+    def _print_selections(self, selections: list[GraphQLSelection]) -> str:
         selections_text = "\n".join(
             [self._print_selection(selection) for selection in selections]
         )

strawberry/codegen/plugins/python.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import textwrap
 from collections import defaultdict
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Dict, List, Optional, Set
+from typing import TYPE_CHECKING, Optional
 
 from strawberry.codegen import CodegenFile, QueryCodegenPlugin
 from strawberry.codegen.types import (
@@ -35,7 +35,7 @@ class PythonType:
 
 
 class PythonPlugin(QueryCodegenPlugin):
-    SCALARS_TO_PYTHON_TYPES: Dict[str, PythonType] = {
+    SCALARS_TO_PYTHON_TYPES: dict[str, PythonType] = {
         "ID": PythonType("str"),
         "Int": PythonType("int"),
         "String": PythonType("str"),
@@ -49,13 +49,13 @@ class PythonPlugin(QueryCodegenPlugin):
     }
 
     def __init__(self, query: Path) -> None:
-        self.imports: Dict[str, Set[str]] = defaultdict(set)
+        self.imports: dict[str, set[str]] = defaultdict(set)
         self.outfile_name: str = query.with_suffix(".py").name
         self.query = query
 
     def generate_code(
-        self, types: List[GraphQLType], operation: GraphQLOperation
-    ) -> List[CodegenFile]:
+        self, types: list[GraphQLType], operation: GraphQLOperation
+    ) -> list[CodegenFile]:
         printed_types = list(filter(None, (self._print_type(type) for type in types)))
         imports = self._print_imports()
 
@@ -80,7 +80,7 @@ class PythonPlugin(QueryCodegenPlugin):
         if isinstance(type_, GraphQLList):
             self.imports["typing"].add("List")
 
-            return f"List[{self._get_type_name(type_.of_type)}]"
+            return f"list[{self._get_type_name(type_.of_type)}]"
 
         if isinstance(type_, GraphQLUnion):
             # TODO: wrong place for this

strawberry/codegen/plugins/typescript.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import textwrap
-from typing import TYPE_CHECKING, List
+from typing import TYPE_CHECKING
 
 from strawberry.codegen import CodegenFile, QueryCodegenPlugin
 from strawberry.codegen.types import (
@@ -40,8 +40,8 @@ class TypeScriptPlugin(QueryCodegenPlugin):
         self.query = query
 
     def generate_code(
-        self, types: List[GraphQLType], operation: GraphQLOperation
-    ) -> List[CodegenFile]:
+        self, types: list[GraphQLType], operation: GraphQLOperation
+    ) -> list[CodegenFile]:
         printed_types = list(filter(None, (self._print_type(type) for type in types)))
 
         return [CodegenFile(self.outfile_name, "\n\n".join(printed_types))]

strawberry/codegen/query_codegen.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from collections.abc import Iterable, Mapping, Sequence
 from dataclasses import MISSING, dataclass
 from enum import Enum
 from functools import cmp_to_key, partial
@@ -8,13 +9,7 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Callable,
-    Iterable,
-    List,
-    Mapping,
     Optional,
-    Sequence,
-    Tuple,
-    Type,
     Union,
     cast,
 )
@@ -114,7 +109,7 @@ class CodegenFile:
 
 @dataclass
 class CodegenResult:
-    files: List[CodegenFile]
+    files: list[CodegenFile]
 
     def to_string(self) -> str:
         return "\n".join(f.content for f in self.files) + "\n"
@@ -144,15 +139,15 @@ class QueryCodegenPlugin:
     def on_end(self, result: CodegenResult) -> None: ...
 
     def generate_code(
-        self, types: List[GraphQLType], operation: GraphQLOperation
-    ) -> List[CodegenFile]:
+        self, types: list[GraphQLType], operation: GraphQLOperation
+    ) -> list[CodegenFile]:
         return []
 
 
 class ConsolePlugin:
     def __init__(self, output_dir: Path) -> None:
         self.output_dir = output_dir
-        self.files_generated: List[Path] = []
+        self.files_generated: list[Path] = []
 
     def before_any_start(self) -> None:
         rich.print(
@@ -238,13 +233,13 @@ def _py_to_graphql_value(obj: Any) -> GraphQLArgumentValue:
 class QueryCodegenPluginManager:
     def __init__(
         self,
-        plugins: List[QueryCodegenPlugin],
+        plugins: list[QueryCodegenPlugin],
         console_plugin: Optional[ConsolePlugin] = None,
     ) -> None:
         self.plugins = plugins
         self.console_plugin = console_plugin
 
-    def _sort_types(self, types: List[GraphQLType]) -> List[GraphQLType]:
+    def _sort_types(self, types: list[GraphQLType]) -> list[GraphQLType]:
         """Sort the types.
 
         t1 < t2 iff t2 has a dependency on t1.
@@ -266,7 +261,7 @@ class QueryCodegenPluginManager:
         return sorted(types, key=cmp_to_key(type_cmp))
 
     def generate_code(
-        self, types: List[GraphQLType], operation: GraphQLOperation
+        self, types: list[GraphQLType], operation: GraphQLOperation
     ) -> CodegenResult:
         result = CodegenResult(files=[])
 
@@ -301,12 +296,12 @@ class QueryCodegen:
     def __init__(
         self,
         schema: Schema,
-        plugins: List[QueryCodegenPlugin],
+        plugins: list[QueryCodegenPlugin],
         console_plugin: Optional[ConsolePlugin] = None,
     ) -> None:
         self.schema = schema
         self.plugin_manager = QueryCodegenPluginManager(plugins, console_plugin)
-        self.types: List[GraphQLType] = []
+        self.types: list[GraphQLType] = []
 
     def run(self, query: str) -> CodegenResult:
         self.plugin_manager.on_start()
@@ -395,7 +390,7 @@ class QueryCodegen:
 
     def _convert_selection_set(
         self, selection_set: Optional[SelectionSetNode]
-    ) -> List[GraphQLSelection]:
+    ) -> list[GraphQLSelection]:
         if selection_set is None:
             return []
 
@@ -442,7 +437,7 @@ class QueryCodegen:
 
     def _convert_arguments(
         self, arguments: Iterable[ArgumentNode]
-    ) -> List[GraphQLArgument]:
+    ) -> list[GraphQLArgument]:
         return [
             GraphQLArgument(argument.name.value, self._convert_value(argument.value))
             for argument in arguments
@@ -450,7 +445,7 @@ class QueryCodegen:
 
     def _convert_directives(
         self, directives: Iterable[DirectiveNode]
-    ) -> List[GraphQLDirective]:
+    ) -> list[GraphQLDirective]:
         return [
             GraphQLDirective(
                 directive.name.value,
@@ -500,7 +495,7 @@ class QueryCodegen:
         self,
         variable_definitions: Optional[Iterable[VariableDefinitionNode]],
         operation_name: str,
-    ) -> Tuple[List[GraphQLVariable], Optional[GraphQLObjectType]]:
+    ) -> tuple[list[GraphQLVariable], Optional[GraphQLObjectType]]:
         if not variable_definitions:
             return [], None
 
@@ -508,7 +503,7 @@ class QueryCodegen:
 
         self._collect_type(type_)
 
-        variables: List[GraphQLVariable] = []
+        variables: list[GraphQLVariable] = []
 
         for variable_definition in variable_definitions:
             variable_type = self._collect_type_from_variable(variable_definition.type)
@@ -522,7 +517,7 @@ class QueryCodegen:
 
         return variables, type_
 
-    def _get_operations(self, ast: DocumentNode) -> List[OperationDefinitionNode]:
+    def _get_operations(self, ast: DocumentNode) -> list[OperationDefinitionNode]:
         return [
             definition
             for definition in ast.definitions
@@ -642,7 +637,7 @@ class QueryCodegen:
 
     def _unwrap_type(
         self, type_: Union[type, StrawberryType]
-    ) -> Tuple[
+    ) -> tuple[
         Union[type, StrawberryType], Optional[Callable[[GraphQLType], GraphQLType]]
     ]:
         wrapper: Optional[Callable[[GraphQLType], GraphQLType]] = None
@@ -777,7 +772,7 @@ class QueryCodegen:
             )
 
         current_type = graph_ql_object_type_factory(class_name)
-        fields: List[Union[GraphQLFragmentSpread, GraphQLField]] = []
+        fields: list[Union[GraphQLFragmentSpread, GraphQLField]] = []
 
         for sub_selection in selection_set.selections:
             if isinstance(sub_selection, FragmentSpreadNode):
@@ -805,7 +800,7 @@ class QueryCodegen:
         # `GraphQLField` or `GraphQLFragmentSpread`
         # and the suite above will cause this statement to be
         # skipped if there are any `GraphQLFragmentSpread`.
-        current_type.fields = cast(List[GraphQLField], fields)
+        current_type.fields = cast(list[GraphQLField], fields)
 
         self._collect_type(current_type)
 
@@ -821,12 +816,12 @@ class QueryCodegen:
         selection: HasSelectionSet,
         parent_type: StrawberryObjectDefinition,
         class_name: str,
-    ) -> List[GraphQLObjectType]:
+    ) -> list[GraphQLObjectType]:
         assert selection.selection_set
 
-        common_fields: List[GraphQLField] = []
-        fragments: List[InlineFragmentNode] = []
-        sub_types: List[GraphQLObjectType] = []
+        common_fields: list[GraphQLField] = []
+        fragments: list[InlineFragmentNode] = []
+        sub_types: list[GraphQLObjectType] = []
 
         for sub_selection in selection.selection_set.selections:
             if isinstance(sub_selection, FieldNode):
@@ -848,7 +843,7 @@ class QueryCodegen:
                 list(common_fields),
                 graphql_typename=type_condition_name,
             )
-            fields: List[Union[GraphQLFragmentSpread, GraphQLField]] = []
+            fields: list[Union[GraphQLFragmentSpread, GraphQLField]] = []
 
             for sub_selection in fragment.selection_set.selections:
                 if isinstance(sub_selection, FragmentSpreadNode):
@@ -893,7 +888,7 @@ class QueryCodegen:
             # `GraphQLField` or `GraphQLFragmentSpread`
             # and the suite above will cause this statement to be
             # skipped if there are any `GraphQLFragmentSpread`.
-            current_type.fields.extend(cast(List[GraphQLField], fields))
+            current_type.fields.extend(cast(list[GraphQLField], fields))
 
             sub_types.append(current_type)
 
@@ -903,7 +898,7 @@ class QueryCodegen:
         return sub_types
 
     def _collect_scalar(
-        self, scalar_definition: ScalarDefinition, python_type: Optional[Type]
+        self, scalar_definition: ScalarDefinition, python_type: Optional[type]
     ) -> GraphQLScalar:
         graphql_scalar = GraphQLScalar(scalar_definition.name, python_type=python_type)
 
@@ -922,9 +917,9 @@ class QueryCodegen:
 
 
 __all__ = [
-    "QueryCodegen",
-    "QueryCodegenPlugin",
-    "ConsolePlugin",
     "CodegenFile",
     "CodegenResult",
+    "ConsolePlugin",
+    "QueryCodegen",
+    "QueryCodegenPlugin",
 ]

strawberry/codegen/types.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, List, Mapping, Optional, Type, Union
+from typing import TYPE_CHECKING, Optional, Union
 
 if TYPE_CHECKING:
+    from collections.abc import Mapping
     from enum import EnumMeta
     from typing_extensions import Literal
 
@@ -23,7 +24,7 @@ class GraphQLList:
 @dataclass
 class GraphQLUnion:
     name: str
-    types: List[GraphQLObjectType]
+    types: list[GraphQLObjectType]
 
 
 @dataclass
@@ -42,7 +43,7 @@ class GraphQLFragmentSpread:
 @dataclass
 class GraphQLObjectType:
     name: str
-    fields: List[GraphQLField] = field(default_factory=list)
+    fields: list[GraphQLField] = field(default_factory=list)
     graphql_typename: Optional[str] = None
 
 
@@ -52,7 +53,7 @@ class GraphQLObjectType:
 @dataclass
 class GraphQLFragmentType(GraphQLObjectType):
     name: str
-    fields: List[GraphQLField] = field(default_factory=list)
+    fields: list[GraphQLField] = field(default_factory=list)
     graphql_typename: Optional[str] = None
     on: str = ""
 
@@ -66,14 +67,14 @@ class GraphQLFragmentType(GraphQLObjectType):
 @dataclass
 class GraphQLEnum:
     name: str
-    values: List[str]
+    values: list[str]
     python_type: EnumMeta
 
 
 @dataclass
 class GraphQLScalar:
     name: str
-    python_type: Optional[Type]
+    python_type: Optional[type]
 
 
 GraphQLType = Union[
@@ -90,15 +91,15 @@ GraphQLType = Union[
 class GraphQLFieldSelection:
     field: str
     alias: Optional[str]
-    selections: List[GraphQLSelection]
-    directives: List[GraphQLDirective]
-    arguments: List[GraphQLArgument]
+    selections: list[GraphQLSelection]
+    directives: list[GraphQLDirective]
+    arguments: list[GraphQLArgument]
 
 
 @dataclass
 class GraphQLInlineFragment:
     type_condition: str
-    selections: List[GraphQLSelection]
+    selections: list[GraphQLSelection]
 
 
 GraphQLSelection = Union[
@@ -141,7 +142,7 @@ class GraphQLNullValue:
 
 @dataclass
 class GraphQLListValue:
-    values: List[GraphQLArgumentValue]
+    values: list[GraphQLArgumentValue]
 
 
 @dataclass
@@ -176,7 +177,7 @@ class GraphQLArgument:
 @dataclass
 class GraphQLDirective:
     name: str
-    arguments: List[GraphQLArgument]
+    arguments: list[GraphQLArgument]
 
 
 @dataclass
@@ -189,39 +190,39 @@ class GraphQLVariable:
 class GraphQLOperation:
     name: str
     kind: Literal["query", "mutation", "subscription"]
-    selections: List[GraphQLSelection]
-    directives: List[GraphQLDirective]
-    variables: List[GraphQLVariable]
+    selections: list[GraphQLSelection]
+    directives: list[GraphQLDirective]
+    variables: list[GraphQLVariable]
     type: GraphQLObjectType
     variables_type: Optional[GraphQLObjectType]
 
 
 __all__ = [
-    "GraphQLOptional",
-    "GraphQLList",
-    "GraphQLUnion",
+    "GraphQLArgument",
+    "GraphQLArgumentValue",
+    "GraphQLBoolValue",
+    "GraphQLDirective",
+    "GraphQLEnum",
+    "GraphQLEnumValue",
     "GraphQLField",
+    "GraphQLFieldSelection",
+    "GraphQLFloatValue",
     "GraphQLFragmentSpread",
-    "GraphQLObjectType",
     "GraphQLFragmentType",
-    "GraphQLEnum",
-    "GraphQLScalar",
-    "GraphQLType",
-    "GraphQLFieldSelection",
     "GraphQLInlineFragment",
-    "GraphQLSelection",
-    "GraphQLStringValue",
     "GraphQLIntValue",
-    "GraphQLFloatValue",
-    "GraphQLEnumValue",
-    "GraphQLBoolValue",
-    "GraphQLNullValue",
+    "GraphQLList",
     "GraphQLListValue",
+    "GraphQLNullValue",
+    "GraphQLObjectType",
     "GraphQLObjectValue",
-    "GraphQLVariableReference",
-    "GraphQLArgumentValue",
-    "GraphQLArgument",
-    "GraphQLDirective",
-    "GraphQLVariable",
     "GraphQLOperation",
+    "GraphQLOptional",
+    "GraphQLScalar",
+    "GraphQLSelection",
+    "GraphQLStringValue",
+    "GraphQLType",
+    "GraphQLUnion",
+    "GraphQLVariable",
+    "GraphQLVariableReference",
 ]

strawberry/codemods/annotated_unions.py

@@ -1,13 +1,16 @@
 from __future__ import annotations
 
-from typing import Optional, Sequence
+from typing import TYPE_CHECKING, Optional
 
 import libcst as cst
 import libcst.matchers as m
-from libcst._nodes.expression import BaseExpression, Call  # noqa: TCH002
+from libcst._nodes.expression import BaseExpression, Call  # noqa: TC002
 from libcst.codemod import CodemodContext, VisitorBasedCodemodCommand
 from libcst.codemod.visitors import AddImportsVisitor, RemoveImportsVisitor
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
 
 def _find_named_argument(args: Sequence[cst.Arg], name: str) -> cst.Arg | None:
     return next(

strawberry/dataloader.py

@@ -8,16 +8,9 @@ from dataclasses import dataclass
 from typing import (
     TYPE_CHECKING,
     Any,
-    Awaitable,
     Callable,
-    Dict,
     Generic,
-    Hashable,
-    Iterable,
-    List,
-    Mapping,
     Optional,
-    Sequence,
     TypeVar,
     Union,
     overload,
@@ -27,6 +20,7 @@ from .exceptions import WrongNumberOfResultsReturned
 
 if TYPE_CHECKING:
     from asyncio.events import AbstractEventLoop
+    from collections.abc import Awaitable, Hashable, Iterable, Mapping, Sequence
 
 
 T = TypeVar("T")
@@ -41,7 +35,7 @@ class LoaderTask(Generic[K, T]):
 
 @dataclass
 class Batch(Generic[K, T]):
-    tasks: List[LoaderTask] = dataclasses.field(default_factory=list)
+    tasks: list[LoaderTask] = dataclasses.field(default_factory=list)
     dispatched: bool = False
 
     def add_task(self, key: Any, future: Future) -> None:
@@ -75,7 +69,7 @@ class DefaultCache(AbstractCache[K, T]):
         self.cache_key_fn: Callable[[K], Hashable] = (
             cache_key_fn if cache_key_fn is not None else lambda x: x
         )
-        self.cache_map: Dict[Hashable, Future[T]] = {}
+        self.cache_map: dict[Hashable, Future[T]] = {}
 
     def get(self, key: K) -> Union[Future[T], None]:
         return self.cache_map.get(self.cache_key_fn(key))
@@ -99,7 +93,7 @@ class DataLoader(Generic[K, T]):
     def __init__(
         self,
         # any BaseException is rethrown in 'load', so should be excluded from the T type
-        load_fn: Callable[[List[K]], Awaitable[Sequence[Union[T, BaseException]]]],
+        load_fn: Callable[[list[K]], Awaitable[Sequence[Union[T, BaseException]]]],
         max_batch_size: Optional[int] = None,
         cache: bool = True,
         loop: Optional[AbstractEventLoop] = None,
@@ -111,7 +105,7 @@ class DataLoader(Generic[K, T]):
     @overload
     def __init__(
         self: DataLoader[K, Any],
-        load_fn: Callable[[List[K]], Awaitable[List[Any]]],
+        load_fn: Callable[[list[K]], Awaitable[list[Any]]],
         max_batch_size: Optional[int] = None,
         cache: bool = True,
         loop: Optional[AbstractEventLoop] = None,
@@ -121,7 +115,7 @@ class DataLoader(Generic[K, T]):
 
     def __init__(
         self,
-        load_fn: Callable[[List[K]], Awaitable[Sequence[Union[T, BaseException]]]],
+        load_fn: Callable[[list[K]], Awaitable[Sequence[Union[T, BaseException]]]],
         max_batch_size: Optional[int] = None,
         cache: bool = True,
         loop: Optional[AbstractEventLoop] = None,
@@ -164,7 +158,7 @@ class DataLoader(Generic[K, T]):
 
         return future
 
-    def load_many(self, keys: Iterable[K]) -> Awaitable[List[T]]:
+    def load_many(self, keys: Iterable[K]) -> Awaitable[list[T]]:
         return gather(*map(self.load, keys))
 
     def clear(self, key: K) -> None:
@@ -210,8 +204,7 @@ class DataLoader(Generic[K, T]):
 def should_create_new_batch(loader: DataLoader, batch: Batch) -> bool:
     return bool(
         batch.dispatched
-        or loader.max_batch_size
-        and len(batch) >= loader.max_batch_size
+        or (loader.max_batch_size and len(batch) >= loader.max_batch_size)
     )
 
 
@@ -267,13 +260,13 @@ async def dispatch_batch(loader: DataLoader, batch: Batch) -> None:
 
 
 __all__ = [
-    "DataLoader",
-    "Batch",
-    "LoaderTask",
     "AbstractCache",
+    "Batch",
+    "DataLoader",
     "DefaultCache",
-    "should_create_new_batch",
-    "get_current_batch",
+    "LoaderTask",
     "dispatch",
     "dispatch_batch",
+    "get_current_batch",
+    "should_create_new_batch",
 ]

strawberry/directive.py

@@ -2,8 +2,15 @@ from __future__ import annotations
 
 import dataclasses
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, Callable, Generic, List, Optional, TypeVar
-from typing_extensions import Annotated
+from typing import (
+    TYPE_CHECKING,
+    Annotated,
+    Any,
+    Callable,
+    Generic,
+    Optional,
+    TypeVar,
+)
 
 from graphql import DirectiveLocation
 
@@ -84,17 +91,17 @@ class StrawberryDirective(Generic[T]):
     python_name: str
     graphql_name: Optional[str]
     resolver: StrawberryDirectiveResolver[T]
-    locations: List[DirectiveLocation]
+    locations: list[DirectiveLocation]
     description: Optional[str] = None
 
     @cached_property
-    def arguments(self) -> List[StrawberryArgument]:
+    def arguments(self) -> list[StrawberryArgument]:
         return self.resolver.arguments
 
 
 def directive(
     *,
-    locations: List[DirectiveLocation],
+    locations: list[DirectiveLocation],
     description: Optional[str] = None,
     name: Optional[str] = None,
 ) -> Callable[[Callable[..., T]], StrawberryDirective[T]]:

strawberry/django/test/client.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from strawberry.test import BaseGraphQLTestClient
 
@@ -6,9 +6,9 @@ from strawberry.test import BaseGraphQLTestClient
 class GraphQLTestClient(BaseGraphQLTestClient):
     def request(
         self,
-        body: Dict[str, object],
-        headers: Optional[Dict[str, object]] = None,
-        files: Optional[Dict[str, object]] = None,
+        body: dict[str, object],
+        headers: Optional[dict[str, object]] = None,
+        files: Optional[dict[str, object]] = None,
     ) -> Any:
         if files:
             return self._client.post(