graphql-core 3.3.0b0__tar.gz → 3.3.0b2__tar.gz

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 3.3.0b0
+current_version = 3.3.0b2
 commit = False
 tag = False
 

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphql-core
-Version: 3.3.0b0
+Version: 3.3.0b2
 Summary: GraphQL-core is a Python port of GraphQL.js, the JavaScript reference implementation for GraphQL.
 Keywords: graphql
 Author: Christoph Zwerschke
@@ -42,8 +42,8 @@ with GraphQL.js.
 The current stable version 3.2.11 of GraphQL-core is up-to-date with GraphQL.js
 version 16.14.1 and supports Python versions 3.7 to 3.14.
 
-You can also try out the latest beta version 3.3.0b0 of GraphQL-core,
-which is up-to-date with GraphQL.js version 17.0.0b0.
+You can also try out the latest beta version 3.3.0b2 of GraphQL-core,
+which is up-to-date with GraphQL.js version 17.0.0b2.
 This new minor version of GraphQL-core also supports Python versions 3.10 to 3.14.
 
 Note that for various reasons, GraphQL-core does not use SemVer like GraphQL.js.

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/README.md

@@ -18,8 +18,8 @@ with GraphQL.js.
 The current stable version 3.2.11 of GraphQL-core is up-to-date with GraphQL.js
 version 16.14.1 and supports Python versions 3.7 to 3.14.
 
-You can also try out the latest beta version 3.3.0b0 of GraphQL-core,
-which is up-to-date with GraphQL.js version 17.0.0b0.
+You can also try out the latest beta version 3.3.0b2 of GraphQL-core,
+which is up-to-date with GraphQL.js version 17.0.0b2.
 This new minor version of GraphQL-core also supports Python versions 3.10 to 3.14.
 
 Note that for various reasons, GraphQL-core does not use SemVer like GraphQL.js.

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/docs/conf.py

@@ -52,7 +52,7 @@ author = "Christoph Zwerschke"
 # The short X.Y version.
 # version = '3.3'
 # The full version, including alpha/beta/rc tags.
-version = release = "3.3.0b0"
+version = release = "3.3.0b2"
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.
@@ -100,7 +100,7 @@ graphql_modules = {
 # be nitpicky
 nitpicky = True
 # but not too nitpicky
-suppress_warnings = ["ref.class", "ref.obj", "ref.python"]
+suppress_warnings = ["ref.attr", "ref.class", "ref.obj", "ref.python"]
 
 # The name of the Pygments (syntax highlighting) style to use.
 pygments_style = "sphinx"

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/docs/modules/execution.rst

@@ -65,6 +65,10 @@ Execution
 
 .. autofunction:: create_source_event_stream
 
+.. autofunction:: map_source_to_response_event
+
+.. autoclass:: RootSelectionSetExecutor
+
 .. autoclass:: Middleware
 
 .. autoclass:: MiddlewareManager
@@ -72,3 +76,5 @@ Execution
 .. autofunction:: get_directive_values
 
 .. autofunction:: get_variable_values
+
+.. autoclass:: VariableValues

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/docs/modules/language.rst

@@ -25,6 +25,7 @@ Each kind of AST node has its own class:
 .. autoclass:: ConstValueNode
 .. autoclass:: DefinitionNode
 .. autoclass:: DirectiveDefinitionNode
+.. autoclass:: DirectiveExtensionNode
 .. autoclass:: DirectiveNode
 .. autoclass:: DocumentNode
 .. autoclass:: EnumTypeDefinitionNode

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/docs/modules/utilities.rst

@@ -127,6 +127,7 @@ Compare two GraphQLSchemas and detect changes:
 .. autoclass:: BreakingChangeType
 .. autoclass:: DangerousChange
 .. autoclass:: DangerousChangeType
+.. autoclass:: SchemaChange
 
 Resolve a schema coordinate to the schema element it refers to:
 

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "graphql-core"
-version = "3.3.0b0"
+version = "3.3.0b2"
 description = "GraphQL-core is a Python port of GraphQL.js, the JavaScript reference implementation for GraphQL."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -32,7 +32,7 @@ test = [
     "pytest>=9,<10",
     "pytest-benchmark>=5.2,<6",
     "pytest-cov>=7.1,<8",
-    "pytest-describe>=3.1,<4",
+    "pytest-describe>=3.2,<4",
     "pytest-timeout>=2.4,<3",
     "pytest-codspeed>=5",
     "tox>=4.55,<5",

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/__init__.py

@@ -168,6 +168,7 @@ from .language import (
     UnionTypeExtensionNode,
     EnumTypeExtensionNode,
     InputObjectTypeExtensionNode,
+    DirectiveExtensionNode,
     SchemaCoordinateNode,
     TypeCoordinateNode,
     MemberCoordinateNode,
@@ -248,6 +249,7 @@ from .utilities import (
     DangerousChangeType,
     SafeChange,
     SafeChangeType,
+    SchemaChange,
     find_breaking_changes,
     find_dangerous_changes,
     find_schema_changes,
@@ -438,6 +440,9 @@ from .validation import (
     specified_rules,
     recommended_rules,
     # Individual validation rules.
+    DeferStreamDirectiveLabel,
+    DeferStreamDirectiveOnRootField,
+    DeferStreamDirectiveOnValidOperationsRule,
     ExecutableDefinitionsRule,
     FieldsOnCorrectTypeRule,
     FragmentsOnCompositeTypesRule,
@@ -456,6 +461,7 @@ from .validation import (
     ProvidedRequiredArgumentsRule,
     ScalarLeafsRule,
     SingleFieldSubscriptionsRule,
+    StreamDirectiveOnListField,
     UniqueArgumentNamesRule,
     UniqueDirectivesPerLocationRule,
     UniqueFragmentNamesRule,
@@ -495,6 +501,7 @@ from .execution import (
     # Types
     AsyncWorkFinishedInfo,
     ExecutionHooks,
+    VariableValues,
     Executor,
     ExecutionResult,
     ExperimentalIncrementalExecutionResults,
@@ -512,7 +519,9 @@ from .execution import (
     # Subscription
     subscribe,
     create_source_event_stream,
+    map_source_to_response_event,
     map_async_iterable,
+    RootSelectionSetExecutor,
     # Middleware
     Middleware,
     MiddlewareManager,
@@ -569,10 +578,14 @@ __all__ = [
     "ConstValueNode",
     "DangerousChange",
     "DangerousChangeType",
+    "DeferStreamDirectiveLabel",
+    "DeferStreamDirectiveOnRootField",
+    "DeferStreamDirectiveOnValidOperationsRule",
     "DefinitionNode",
     "DirectiveArgumentCoordinateNode",
     "DirectiveCoordinateNode",
     "DirectiveDefinitionNode",
+    "DirectiveExtensionNode",
     "DirectiveLocation",
     "DirectiveNode",
     "DocumentNode",
@@ -740,12 +753,14 @@ __all__ = [
     "ResolvedNamedType",
     "ResolvedSchemaElement",
     "ResponsePath",
+    "RootSelectionSetExecutor",
     "SDLValidationRule",
     "SafeChange",
     "SafeChangeType",
     "ScalarLeafsRule",
     "ScalarTypeDefinitionNode",
     "ScalarTypeExtensionNode",
+    "SchemaChange",
     "SchemaCoordinateNode",
     "SchemaDefinitionNode",
     "SchemaExtensionNode",
@@ -755,6 +770,7 @@ __all__ = [
     "SingleFieldSubscriptionsRule",
     "Source",
     "SourceLocation",
+    "StreamDirectiveOnListField",
     "StringValueNode",
     "SubsequentIncrementalExecutionResult",
     "Thunk",
@@ -795,6 +811,7 @@ __all__ = [
     "ValuesOfCorrectTypeRule",
     "VariableDefinitionNode",
     "VariableNode",
+    "VariableValues",
     "VariablesAreInputTypesRule",
     "VariablesInAllowedPositionRule",
     "Visitor",
@@ -904,6 +921,7 @@ __all__ = [
     "lexicographic_sort_schema",
     "located_error",
     "map_async_iterable",
+    "map_source_to_response_event",
     "parse",
     "parse_const_value",
     "parse_schema_coordinate",

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/error/graphql_error.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 from sys import exc_info
 from typing import TYPE_CHECKING, Any, TypeAlias, TypedDict
 
+from ..pyutils import Undefined
+
 if TYPE_CHECKING:
     from collections.abc import Collection
 
@@ -108,12 +110,26 @@ class GraphQLError(Exception):
     """
 
     original_error: Exception | None
-    """The original error thrown from a field resolver during execution"""
+    """Original error that caused this GraphQLError, if one exists
+
+    .. deprecated:: 3.3
+        Use :attr:`cause` instead to better align with the standard exception
+        :attr:`~BaseException.__cause__`.
+    """
+
+    cause: Any
+    """The cause of this error, if one exists
+
+    The value passed as the ``cause`` when creating this error. Adopts the
+    convention of the standard exception cause and is also exposed as the
+    :attr:`~BaseException.__cause__` when it is an exception.
+    """
 
     extensions: GraphQLErrorExtensions | None
     """Extension fields to add to the formatted error"""
 
     __slots__ = (
+        "cause",
         "extensions",
         "locations",
         "message",
@@ -135,15 +151,27 @@ class GraphQLError(Exception):
         path: Collection[str | int] | None = None,
         original_error: Exception | None = None,
         extensions: GraphQLErrorExtensions | None = None,
+        cause: Any = Undefined,
     ) -> None:
         """Initialize a GraphQLError."""
         super().__init__(message)
         self.message = message
 
+        # The cause defaults to the (deprecated) original error when not given.
+        has_cause = cause is not Undefined
+        self.cause = cause if has_cause else original_error
+
         if path and not isinstance(path, list):
             path = list(path)
         self.path = path or None  # type: ignore
-        self.original_error = original_error
+
+        # An Error provided as the cause is also exposed as the original error
+        # for backward compatibility.
+        cause_value = cause if has_cause else None
+        underlying_error = original_error
+        if underlying_error is None and isinstance(cause_value, Exception):
+            underlying_error = cause_value
+        self.original_error = underlying_error
 
         # Compute list of blame nodes.
         if nodes and not isinstance(nodes, list):
@@ -170,16 +198,25 @@ class GraphQLError(Exception):
             locations = [loc.source.get_location(loc.start) for loc in node_locations]
         self.locations = locations or None
 
+        # Adopt the cause as the standard exception cause when it is one, so
+        # that the cause chain is reported without duplicating the traceback.
+        if isinstance(cause_value, BaseException):
+            self.__cause__ = cause_value
+
+        # Preserve the traceback of an explicit original error. The traceback of
+        # a cause is not copied over, since Python already reports cause chains.
         if original_error:
             self.__traceback__ = original_error.__traceback__
-            if original_error.__cause__:
-                self.__cause__ = original_error.__cause__
-            elif original_error.__context__:
-                self.__context__ = original_error.__context__
-            if extensions is None:
-                original_extensions = getattr(original_error, "extensions", None)
-                if isinstance(original_extensions, dict):
-                    extensions = original_extensions
+            if not self.__cause__:
+                if original_error.__cause__:
+                    self.__cause__ = original_error.__cause__
+                elif original_error.__context__:
+                    self.__context__ = original_error.__context__
+
+        if extensions is None and underlying_error is not None:
+            original_extensions = getattr(underlying_error, "extensions", None)
+            if isinstance(original_extensions, dict):
+                extensions = original_extensions
         self.extensions = extensions or {}
         if not self.__traceback__:
             self.__traceback__ = exc_info()[2]
@@ -217,7 +254,7 @@ class GraphQLError(Exception):
             and all(
                 getattr(self, slot) == getattr(other, slot)
                 for slot in self.__slots__
-                if slot != "original_error"
+                if slot not in ("cause", "original_error")
             )
         ) or (
             isinstance(other, dict)
@@ -225,7 +262,7 @@ class GraphQLError(Exception):
             and all(
                 slot in self.__slots__ and getattr(self, slot) == other.get(slot)
                 for slot in other
-                if slot != "original_error"
+                if slot not in ("cause", "original_error")
             )
         )
 

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/execution/__init__.py

@@ -30,7 +30,12 @@ from .types import (
     StreamItemsResult,
 )
 from .middleware import MiddlewareManager
-from .values import get_argument_values, get_directive_values, get_variable_values
+from .values import (
+    get_argument_values,
+    get_directive_values,
+    get_variable_values,
+    VariableValues,
+)
 from .execute import (
     create_source_event_stream,
     execute,
@@ -40,12 +45,14 @@ from .execute import (
     execute_sync,
     default_field_resolver,
     default_type_resolver,
+    map_source_to_response_event,
     subscribe,
     AsyncWorkFinishedInfo,
     ExecutionHooks,
     Executor,
     GraphQLWrappedResult,
     Middleware,
+    RootSelectionSetExecutor,
 )
 
 __all__ = [
@@ -72,11 +79,13 @@ __all__ = [
     "Middleware",
     "MiddlewareManager",
     "PendingResult",
+    "RootSelectionSetExecutor",
     "StreamItemRecord",
     "StreamItemResult",
     "StreamItemsResult",
     "StreamRecord",
     "SubsequentIncrementalExecutionResult",
+    "VariableValues",
     "create_source_event_stream",
     "default_field_resolver",
     "default_type_resolver",
@@ -89,5 +98,6 @@ __all__ = [
     "get_directive_values",
     "get_variable_values",
     "map_async_iterable",
+    "map_source_to_response_event",
     "subscribe",
 ]

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/execution/execute.py

@@ -140,6 +140,7 @@ __all__ = [
     "Executor",
     "GraphQLWrappedResult",
     "Middleware",
+    "RootSelectionSetExecutor",
     "create_source_event_stream",
     "default_field_resolver",
     "default_type_resolver",
@@ -148,6 +149,7 @@ __all__ = [
     "execute_subscription_event",
     "execute_sync",
     "experimental_execute_incrementally",
+    "map_source_to_response_event",
     "subscribe",
 ]
 
@@ -237,7 +239,6 @@ class Executor(IncrementalPublisherContext):
     field_resolver: GraphQLFieldResolver
     type_resolver: GraphQLTypeResolver
     subscribe_field_resolver: GraphQLFieldResolver
-    per_event_executor: Callable[[Executor], AwaitableOrValue[ExecutionResult]]
     enable_early_execution: bool
     hide_suggestions: bool
     abort_signal: AbortSignal | None
@@ -274,8 +275,6 @@ class Executor(IncrementalPublisherContext):
         middleware_manager: MiddlewareManager | None = None,
         is_awaitable: Callable[[Any], TypeGuard[Awaitable]] | None = None,
         is_async_iterable: Callable[[Any], TypeGuard[AsyncIterable]] | None = None,
-        per_event_executor: Callable[[Executor], AwaitableOrValue[ExecutionResult]]
-        | None = None,
         hide_suggestions: bool = False,
         abort_signal: AbortSignal | None = None,
         hooks: ExecutionHooks | None = None,
@@ -290,7 +289,6 @@ class Executor(IncrementalPublisherContext):
         self.field_resolver = field_resolver
         self.type_resolver = type_resolver
         self.subscribe_field_resolver = subscribe_field_resolver
-        self.per_event_executor = per_event_executor or execute_subscription_event
         self.enable_early_execution = enable_early_execution
         self.hide_suggestions = hide_suggestions
         self.abort_signal = abort_signal
@@ -331,8 +329,6 @@ class Executor(IncrementalPublisherContext):
         middleware: Middleware | None = None,
         is_awaitable: Callable[[Any], TypeGuard[Awaitable]] | None = None,
         is_async_iterable: Callable[[Any], TypeGuard[AsyncIterable]] | None = None,
-        per_event_executor: Callable[[Executor], AwaitableOrValue[ExecutionResult]]
-        | None = None,
         hide_suggestions: bool = False,
         abort_signal: AbortSignal | None = None,
         hooks: ExecutionHooks | None = None,
@@ -425,7 +421,6 @@ class Executor(IncrementalPublisherContext):
             middleware_manager,
             is_awaitable,
             is_async_iterable,
-            per_event_executor=per_event_executor,
             hide_suggestions=hide_suggestions,
             abort_signal=abort_signal,
             hooks=hooks,
@@ -1978,39 +1973,6 @@ class Executor(IncrementalPublisherContext):
         execution_plans[original_grouped_field_set] = exeecution_plan
         return exeecution_plan
 
-    def map_source_to_response(
-        self, result_or_stream: ExecutionResult | AsyncIterable[Any]
-    ) -> AsyncGenerator[ExecutionResult, None] | ExecutionResult:
-        """Map source result to response.
-
-        For each payload yielded from a subscription,
-        map it over the normal GraphQL :func:`~graphql.execution.execute` function,
-        with ``payload`` as the ``root_value``.
-        This implements the "MapSourceToResponseEvent" algorithm
-        described in the GraphQL specification.
-        Each event is executed with the executor's ``per_event_executor``, which
-        defaults to :func:`~graphql.execution.execute_subscription_event` (providing
-        the "ExecuteSubscriptionEvent" algorithm) but can be overridden to set up and
-        tear down a custom executor around the execution of each event.
-        """
-        if not self.is_async_iterable(result_or_stream):
-            return cast("ExecutionResult", result_or_stream)  # pragma: no cover
-
-        build_executor = self.build_per_event_executor
-        per_event_executor = self.per_event_executor
-
-        async def callback(payload: Any) -> ExecutionResult:
-            result = per_event_executor(build_executor(payload))
-            # typecast to ExecutionResult, not possible to return
-            # ExperimentalIncrementalExecutionResults when operation is 'subscription'.
-            return (
-                await cast("Awaitable[ExecutionResult]", result)
-                if self.is_awaitable(result)
-                else cast("ExecutionResult", result)
-            )
-
-        return map_async_iterable(self.cancellable_iterable(result_or_stream), callback)
-
     def collect_execution_groups(
         self,
         parent_type: GraphQLObjectType,
@@ -2924,8 +2886,6 @@ def subscribe(
     enable_early_execution: bool = False,
     executor_class: type[Executor] | None = None,
     middleware: MiddlewareManager | None = None,
-    per_event_executor: Callable[[Executor], AwaitableOrValue[ExecutionResult]]
-    | None = None,
     hide_suggestions: bool = False,
     **custom_context_args: Any,
 ) -> AwaitableOrValue[AsyncIterator[ExecutionResult] | ExecutionResult]:
@@ -2952,10 +2912,12 @@ def subscribe(
     If an operation that defers or streams data is executed with this function,
     a field error will be raised at the location of the `@defer` or `@stream` directive.
 
-    A custom ``per_event_executor`` may be provided to execute each subscription event
-    with a custom executor. It receives the per-event executor and should
-    return an ExecutionResult, usually by calling
-    :func:`~graphql.execution.execute_subscription_event` (the default executor).
+    To customize how each subscription event is executed, compose the subscription
+    pipeline directly instead of calling this function: build an executor with
+    :meth:`Executor.build`, resolve the source event stream with
+    :func:`~graphql.execution.create_source_event_stream`, and map it to the response
+    stream with :func:`~graphql.execution.map_source_to_response_event`, passing a
+    custom ``root_selection_set_executor``.
     """
     if executor_class is None:
         executor_class = Executor
@@ -2975,7 +2937,6 @@ def subscribe(
         max_coercion_errors,
         enable_early_execution,
         middleware=middleware,
-        per_event_executor=per_event_executor,
         hide_suggestions=hide_suggestions,
         **custom_context_args,
     )
@@ -2994,16 +2955,19 @@ def subscribe(
 
         async def await_result() -> Any:
             awaited_result_or_stream = await result_or_stream
-            if isinstance(awaited_result_or_stream, ExecutionResult):
-                return awaited_result_or_stream
-            return executor.map_source_to_response(awaited_result_or_stream)
+            return (
+                map_source_to_response_event(executor, awaited_result_or_stream)
+                if executor.is_async_iterable(awaited_result_or_stream)
+                else awaited_result_or_stream
+            )
 
         return await_result()
 
-    if isinstance(result_or_stream, ExecutionResult):
-        return result_or_stream
-
-    return executor.map_source_to_response(result_or_stream)  # type: ignore
+    return (
+        map_source_to_response_event(executor, result_or_stream)  # type: ignore
+        if executor.is_async_iterable(result_or_stream)
+        else result_or_stream
+    )
 
 
 def execute_root_selection_set(
@@ -3023,10 +2987,11 @@ def execute_subscription_event(
 ) -> AwaitableOrValue[ExecutionResult]:
     """Execute a single subscription event.
 
-    This is the default ``per_event_executor`` used by :func:`subscribe`. It provides
-    the "ExecuteSubscriptionEvent" algorithm described in the GraphQL specification,
-    which is nearly identical to the "ExecuteQuery" algorithm. A custom executor may
-    wrap this function to set up and tear down a per-event executor.
+    This is the default ``root_selection_set_executor`` used by
+    :func:`map_source_to_response_event`. It provides the "ExecuteSubscriptionEvent"
+    algorithm described in the GraphQL specification, which is nearly identical to the
+    "ExecuteQuery" algorithm. A custom executor may wrap this function to set up and
+    tear down a per-event executor.
 
     The passed executor should be a per-event executor as created by
     :meth:`Executor.build_per_event_executor`.
@@ -3034,6 +2999,47 @@ def execute_subscription_event(
     return cast("AwaitableOrValue[ExecutionResult]", executor.execute_operation(False))
 
 
+RootSelectionSetExecutor: TypeAlias = Callable[
+    ["Executor"], AwaitableOrValue[ExecutionResult]
+]
+
+
+def map_source_to_response_event(
+    executor: Executor,
+    source_event_stream: AsyncIterable[Any],
+    root_selection_set_executor: RootSelectionSetExecutor = execute_subscription_event,
+) -> AsyncGenerator[ExecutionResult, None]:
+    """Map a subscription source event stream to a response event stream.
+
+    Implements the "MapSourceToResponseEvent" algorithm described in the GraphQL
+    specification, mapping each event from a subscription source event stream to an
+    ExecutionResult in the response stream.
+
+    For each payload yielded from the source event stream, it is mapped over the normal
+    GraphQL :func:`~graphql.execution.execute` function, with ``payload`` as the
+    ``root_value``. Each event is executed with the given
+    ``root_selection_set_executor``, which defaults to
+    :func:`~graphql.execution.execute_subscription_event` (providing the
+    "ExecuteSubscriptionEvent" algorithm) but can be overridden to set up and tear
+    down a custom executor around the execution of each event.
+    """
+    build_executor = executor.build_per_event_executor
+
+    async def callback(payload: Any) -> ExecutionResult:
+        result = root_selection_set_executor(build_executor(payload))
+        # typecast to ExecutionResult, not possible to return
+        # ExperimentalIncrementalExecutionResults when operation is 'subscription'.
+        return (
+            await cast("Awaitable[ExecutionResult]", result)
+            if executor.is_awaitable(result)
+            else cast("ExecutionResult", result)
+        )
+
+    return map_async_iterable(
+        executor.cancellable_iterable(source_event_stream), callback
+    )
+
+
 def create_source_event_stream(
     executor: Executor,
 ) -> AwaitableOrValue[AsyncIterable[Any] | ExecutionResult]:

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/execution/values.py

@@ -6,6 +6,8 @@ from typing import TYPE_CHECKING, Any, NamedTuple, TypeAlias
 
 from ..error import GraphQLError
 from ..language import (
+    DirectiveDefinitionNode,
+    DirectiveExtensionNode,
     DirectiveNode,
     EnumValueDefinitionNode,
     ExecutableDefinitionNode,
@@ -425,7 +427,9 @@ def print_argument_or_fragment_variable(
 
 
 NodeWithDirective: TypeAlias = (
-    EnumValueDefinitionNode
+    DirectiveDefinitionNode
+    | DirectiveExtensionNode
+    | EnumValueDefinitionNode
     | ExecutableDefinitionNode
     | FieldDefinitionNode
     | InputValueDefinitionNode

{graphql_core-3.3.0b0 → graphql_core-3.3.0b2}/src/graphql/language/__init__.py

@@ -101,6 +101,7 @@ from .ast import (
     UnionTypeExtensionNode,
     EnumTypeExtensionNode,
     InputObjectTypeExtensionNode,
+    DirectiveExtensionNode,
     SchemaCoordinateNode,
     TypeCoordinateNode,
     MemberCoordinateNode,
@@ -142,6 +143,7 @@ __all__ = [
     "DirectiveArgumentCoordinateNode",
     "DirectiveCoordinateNode",
     "DirectiveDefinitionNode",
+    "DirectiveExtensionNode",
     "DirectiveLocation",
     "DirectiveNode",
     "DocumentNode",