strawberry-graphql 0.173.1__py3-none-any.whl → 0.174.0__py3-none-any.whl

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/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/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/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/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]

{strawberry_graphql-0.173.1.dist-info → strawberry_graphql-0.174.0.dist-info}/METADATA

@@ -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.dist-info → strawberry_graphql-0.174.0.dist-info}/RECORD

@@ -88,7 +88,7 @@ strawberry/ext/dataclasses/LICENSE,sha256=WZgm35K_3NJwLqxpEHJJi7CWxVrwTumEz5D3Dt
 strawberry/ext/dataclasses/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/ext/dataclasses/dataclasses.py,sha256=znJ1R_TwZDO3lvDeMiy0YijFK8e6i3OkFMdtSdvf83k,2175
 strawberry/ext/mypy_plugin.py,sha256=Z7U2gguKjnDdggiAP5by-Xa4hbmvgLx37zkvDewsvpY,36343
-strawberry/extensions/__init__.py,sha256=48dT5z6VXSJ4Uh7UKyMrhE49090TrZEjofJlOWFiPaU,1062
+strawberry/extensions/__init__.py,sha256=DS0SPbZVhiFTIsCuZqQT4ibxlCMZ6TALq3qTwrqaKrs,1127
 strawberry/extensions/add_validation_rules.py,sha256=U-_U3aTNGqWd88zbzqedrLGjP1R-Pv1KyF8x_hIOoEw,1358
 strawberry/extensions/base_extension.py,sha256=Cz_ez807YB9CNe_9KrikrbJ_1WMADjGiXdwCbmW2P3s,1773
 strawberry/extensions/context.py,sha256=hBCbOEzf3S0V3Q0-xj9CaLKgO-rSVrtjGtl_JoTnvgg,6964
@@ -96,7 +96,8 @@ strawberry/extensions/directives.py,sha256=UqdH3wmK0z0TfE6bYDsBitEVdicr3qCzjzbzF
 strawberry/extensions/disable_validation.py,sha256=eLlZ0SEuvK3HUSvLm4PpdgrtP30ckG25sV3hGd2W-sM,720
 strawberry/extensions/field_extension.py,sha256=vvcL4I9BXkoVgm5nojA0TFqqOMxG6LxeKGrb9sa8H_c,5584
 strawberry/extensions/mask_errors.py,sha256=5uClsvhuRjsvU4Va9eHoWn5ePGuoWXrEudEIaS41sp8,1385
-strawberry/extensions/max_aliases.py,sha256=pDEoLnXBThgjurwbypme-UH6JZ4n1_jqyRQwPP_2u_0,2142
+strawberry/extensions/max_aliases.py,sha256=turajK8BMOmp59o0Wr4MwyISrkn17XJmGQPVCwNkXjk,2142
+strawberry/extensions/max_tokens.py,sha256=Fa-Zcugo2Y1WRINTdTRWzsRhAnBFCLAoplfnlS8j_nY,1014
 strawberry/extensions/parser_cache.py,sha256=pyZ71_oeEn0YLAZKZYidOIzpBPKp570l8GBQvW-VuN4,1186
 strawberry/extensions/pyinstrument.py,sha256=Ac21atKEyOuZs1s_1-NEes3vB8QlM5ZGw8xU6EMdMPM,892
 strawberry/extensions/query_depth_limiter.py,sha256=_Q2AI2iE51dCCTqFMl8henaHR_GEc51sPA4xviQdsGM,7852
@@ -162,7 +163,7 @@ strawberry/schema/base.py,sha256=rj4pMnVSdTclZpYIPoyWlwEtSYohqJPJzEgInsoON08,289
 strawberry/schema/compat.py,sha256=fEOig4-X5Q3TrbuHcWXk1t69TzMfOr8vy3U-xB03yEU,1949
 strawberry/schema/config.py,sha256=G1MC5DNKvSFNmg4bcu9XjQhceW8xZMIHf1hG3aS4SGA,599
 strawberry/schema/exceptions.py,sha256=_9Ua-lLRCJfbtd8B6MXILNKGI2SwHkcBNkAHYM7ITp8,534
-strawberry/schema/execute.py,sha256=vUjrEGzbO5L4YMIosx3RTVt9TQyogf6AeBbKWoMFdO0,10085
+strawberry/schema/execute.py,sha256=6U-JOTtYn6qQgEpHpJvqkm9z5n5IwsdCSjy8yFgMVcw,10295
 strawberry/schema/name_converter.py,sha256=eQh4A-8fQAkhwgFhGk0_qFUQ2NPfqKO8DzKJVrxf0vk,6284
 strawberry/schema/schema.py,sha256=X_i3TE0Vvm9AV9QrV9qDBdSisbxAkpajpv1RXmQoYG8,11915
 strawberry/schema/schema_converter.py,sha256=eCGqb9TA3B8ipByB9Dv5bqfZ_yY4UerhbQcZXts4UhQ,30619
@@ -191,7 +192,7 @@ strawberry/tools/create_type.py,sha256=Hv2bk6uFsNeBqqXp8egA6vYzxCdudSsqpZRE1M_lk
 strawberry/tools/merge_types.py,sha256=31AqGSEn-z5FhMNUEyfFNovX_w_WGHWd9WxtXJ8X-w0,957
 strawberry/type.py,sha256=Z8WVkD_hm6qfsUSodAb4ILpOjjGQB_rz88josm0WlfM,4441
 strawberry/types/__init__.py,sha256=APb1Cjy6bxqFxIfDfempP6eb9NE3LYDwQ3gX7r07lXI,139
-strawberry/types/execution.py,sha256=fKvwWwOeB6zBXMyPh8GCApeL5HsJZOsqPOyeYifL9dc,2476
+strawberry/types/execution.py,sha256=wPXvgaWDaF1o_qJ3WzqpVOp64_uGbB87nbyL69Gdy_4,2768
 strawberry/types/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/types/fields/resolver.py,sha256=gvhjPbBhjaZVPe5jbt61hgKwpMIutC3IzKTMi8JgaYg,12010
 strawberry/types/graphql.py,sha256=3SWZEsa0Zy1eVW6vy75BnB7t9_lJVi6TBV3_1j3RNBs,687
@@ -214,8 +215,8 @@ strawberry/utils/logging.py,sha256=pz1pRJtx5xX2ejGKNN-fG8zPA4SJjnq1HK9xaQpjd4s,1
 strawberry/utils/operation.py,sha256=Um-tBCPl3_bVFN2Ph7o1mnrxfxBes4HFCj6T0x4kZxE,1135
 strawberry/utils/str_converters.py,sha256=VdOnzaxhwJnmQl1Lon0VOjl72uXhM8tLfGxoGwn3arI,657
 strawberry/utils/typing.py,sha256=KtfgwoTTNYSw6IDZWf1McwzxZs5D1guNCVmXt8Mh9z8,10071
-strawberry_graphql-0.173.1.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.173.1.dist-info/METADATA,sha256=yO2dD0i9N5Q9M3D2buMwmfC42-kIidof-_9ovcoMeb0,7644
-strawberry_graphql-0.173.1.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-strawberry_graphql-0.173.1.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.173.1.dist-info/RECORD,,
+strawberry_graphql-0.174.0.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.174.0.dist-info/METADATA,sha256=qdjW1II07Hr3SeMHtNfr7FN26QgItKxcWrZJc0L_PIw,7644
+strawberry_graphql-0.174.0.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
+strawberry_graphql-0.174.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.174.0.dist-info/RECORD,,

{strawberry_graphql-0.173.1.dist-info → strawberry_graphql-0.174.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.5.2
+Generator: poetry-core 1.5.1
 Root-Is-Purelib: true
 Tag: py3-none-any