strawberry-graphql 0.173.1__tar.gz → 0.174.0__tar.gz

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/PKG-INFO

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

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "strawberry-graphql"
 packages = [ { include = "strawberry" } ]
-version = "0.173.1"
+version = "0.174.0"
 description = "A library for creating GraphQL APIs"
 authors = ["Patrick Arminio <patrick.arminio@gmail.com>"]
 license = "MIT"
@@ -154,7 +154,8 @@ markers = [
     "fastapi",
     "chalice",
     "flask",
-    "starlite"
+    "starlite",
+    "flaky"
 ]
 asyncio_mode = "auto"
 filterwarnings = [

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/strawberry/extensions/__init__.py

@@ -6,6 +6,7 @@ from .disable_validation import DisableValidation
 from .field_extension import FieldExtension
 from .mask_errors import MaskErrors
 from .max_aliases import MaxAliasesLimiter
+from .max_tokens import MaxTokensLimiter
 from .parser_cache import ParserCache
 from .query_depth_limiter import QueryDepthLimiter
 from .validation_cache import ValidationCache
@@ -36,4 +37,5 @@ __all__ = [
     "ValidationCache",
     "MaskErrors",
     "MaxAliasesLimiter",
+    "MaxTokensLimiter",
 ]

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/strawberry/extensions/max_aliases.py

@@ -19,7 +19,7 @@ class MaxAliasesLimiter(AddValidationRules):
     Example:
 
     >>> import strawberry
-    >>> from strawberry.extensions import QueryDepthLimiter
+    >>> from strawberry.extensions import MaxAliasesLimiter
     >>>
     >>> schema = strawberry.Schema(
     ...     Query,

strawberry_graphql-0.174.0/strawberry/extensions/max_tokens.py

@@ -0,0 +1,44 @@
+from typing import Iterator
+
+from strawberry.extensions.base_extension import SchemaExtension
+
+
+class MaxTokensLimiter(SchemaExtension):
+    """
+    Add a validator to limit the number of tokens in a GraphQL document.
+
+    Example:
+
+    >>> import strawberry
+    >>> from strawberry.extensions import MaxTokensLimiter
+    >>>
+    >>> schema = strawberry.Schema(
+    ...     Query,
+    ...     extensions=[
+    ...         MaxTokensLimiter(max_token_count=1000)
+    ...     ]
+    ... )
+
+    Arguments:
+
+    `max_token_count: int`
+        The maximum number of tokens allowed in a GraphQL document.
+
+    The following things are counted as tokens:
+    * various brackets: "{", "}", "(", ")"
+    * colon :
+    * words
+
+    Not counted:
+    * quotes
+    """
+
+    def __init__(
+        self,
+        max_token_count: int,
+    ):
+        self.max_token_count = max_token_count
+
+    def on_operation(self) -> Iterator[None]:
+        self.execution_context.parse_options["max_tokens"] = self.max_token_count
+        yield

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/strawberry/schema/execute.py

@@ -27,6 +27,8 @@ from strawberry.types import ExecutionResult
 from .exceptions import InvalidOperationTypeError
 
 if TYPE_CHECKING:
+    from typing_extensions import Unpack
+
     from graphql import ExecutionContext as GraphQLExecutionContext
     from graphql import ExecutionResult as GraphQLExecutionResult
     from graphql import GraphQLSchema
@@ -35,11 +37,12 @@ if TYPE_CHECKING:
 
     from strawberry.extensions import SchemaExtension
     from strawberry.types import ExecutionContext
+    from strawberry.types.execution import ParseOptions
     from strawberry.types.graphql import OperationType
 
 
-def parse_document(query: str) -> DocumentNode:
-    return parse(query)
+def parse_document(query: str, **kwargs: Unpack[ParseOptions]) -> DocumentNode:
+    return parse(query, **kwargs)
 
 
 def validate_document(
@@ -90,7 +93,7 @@ async def execute(
             try:
                 if not execution_context.graphql_document:
                     execution_context.graphql_document = parse_document(
-                        execution_context.query
+                        execution_context.query, **execution_context.parse_options
                     )
 
             except GraphQLError as error:
@@ -183,7 +186,7 @@ def execute_sync(
             try:
                 if not execution_context.graphql_document:
                     execution_context.graphql_document = parse_document(
-                        execution_context.query
+                        execution_context.query, **execution_context.parse_options
                     )
 
             except GraphQLError as error:

{strawberry_graphql-0.173.1 → strawberry_graphql-0.174.0}/strawberry/types/execution.py

@@ -1,13 +1,24 @@
 from __future__ import annotations
 
 import dataclasses
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    List,
+    Optional,
+    Tuple,
+    Type,
+)
+from typing_extensions import TypedDict
 
 from graphql import specified_rules
 
 from strawberry.utils.operation import get_first_operation, get_operation_type
 
 if TYPE_CHECKING:
+    from typing_extensions import NotRequired
+
     from graphql import ASTValidationRule
     from graphql import ExecutionResult as GraphQLExecutionResult
     from graphql.error.graphql_error import GraphQLError
@@ -24,6 +35,9 @@ class ExecutionContext:
     schema: Schema
     context: Any = None
     variables: Optional[Dict[str, Any]] = None
+    parse_options: ParseOptions = dataclasses.field(
+        default_factory=lambda: ParseOptions()
+    )
     root_value: Optional[Any] = None
     validation_rules: Tuple[Type[ASTValidationRule], ...] = dataclasses.field(
         default_factory=lambda: tuple(specified_rules)
@@ -76,3 +90,7 @@ class ExecutionResult:
     data: Optional[Dict[str, Any]]
     errors: Optional[List[GraphQLError]]
     extensions: Optional[Dict[str, Any]] = None
+
+
+class ParseOptions(TypedDict):
+    max_tokens: NotRequired[int]