Use postponed evaluation of type annotations

Author: Cito

src/graphql/error/syntax_error.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from typing import TYPE_CHECKING
 
 from .graphql_error import GraphQLError
@@ -11,7 +13,7 @@
 class GraphQLSyntaxError(GraphQLError):
     """A GraphQLError representing a syntax error."""
 
-    def __init__(self, source: "Source", position: int, description: str) -> None:
+    def __init__(self, source: Source, position: int, description: str) -> None:
         super().__init__(
             f"Syntax Error: {description}", source=source, positions=[position]
         )

src/graphql/execution/map_async_iterator.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from asyncio import CancelledError, Event, Task, ensure_future, wait
 from concurrent.futures import FIRST_COMPLETED
 from inspect import isasyncgen, isawaitable
@@ -23,7 +25,7 @@ def __init__(self, iterable: AsyncIterable, callback: Callable) -> None:
         self.callback = callback
         self._close_event = Event()
 
-    def __aiter__(self) -> "MapAsyncIterator":
+    def __aiter__(self) -> MapAsyncIterator:
         """Get the iterator object."""
         return self
 

src/graphql/language/ast.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from copy import copy, deepcopy
 from enum import Enum
 from typing import Any, Dict, List, Tuple, Optional, Union
@@ -137,7 +139,7 @@ def __hash__(self) -> int:
             (self.kind, self.start, self.end, self.line, self.column, self.value)
         )
 
-    def __copy__(self) -> "Token":
+    def __copy__(self) -> Token:
         """Create a shallow copy of the token"""
         token = self.__class__(
             self.kind,
@@ -150,7 +152,7 @@ def __copy__(self) -> "Token":
         token.prev = self.prev
         return token
 
-    def __deepcopy__(self, memo: Dict) -> "Token":
+    def __deepcopy__(self, memo: Dict) -> Token:
         """Allow only shallow copies to avoid recursion."""
         return copy(self)
 
@@ -360,11 +362,11 @@ def __setattr__(self, key: str, value: Any) -> None:
             del self._hash
         super().__setattr__(key, value)
 
-    def __copy__(self) -> "Node":
+    def __copy__(self) -> Node:
         """Create a shallow copy of the node."""
         return self.__class__(**{key: getattr(self, key) for key in self.keys})
 
-    def __deepcopy__(self, memo: Dict) -> "Node":
+    def __deepcopy__(self, memo: Dict) -> Node:
         """Create a deep copy of the node"""
         # noinspection PyArgumentList
         return self.__class__(

src/graphql/language/location.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from typing import Any, NamedTuple, TYPE_CHECKING
 
 try:
@@ -37,7 +39,7 @@ def __ne__(self, other: Any) -> bool:
         return not self == other
 
 
-def get_location(source: "Source", position: int) -> SourceLocation:
+def get_location(source: Source, position: int) -> SourceLocation:
     """Get the line and column for a character position in the source.
 
     Takes a Source and a UTF-8 character offset, and returns the corresponding line and

src/graphql/pyutils/frozen_dict.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from copy import deepcopy
 from typing import Dict, TypeVar
 
@@ -28,12 +30,12 @@ def __iadd__(self, value):
     def __hash__(self):
         return hash(tuple(self.items()))
 
-    def __copy__(self) -> "FrozenDict":
+    def __copy__(self) -> FrozenDict:
         return FrozenDict(self)
 
     copy = __copy__
 
-    def __deepcopy__(self, memo: Dict) -> "FrozenDict":
+    def __deepcopy__(self, memo: Dict) -> FrozenDict:
         return FrozenDict({k: deepcopy(v, memo) for k, v in self.items()})
 
     def clear(self):

src/graphql/pyutils/frozen_list.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from copy import deepcopy
 from typing import Dict, List, TypeVar
 
@@ -39,10 +41,10 @@ def __imul__(self, value):
     def __hash__(self):
         return hash(tuple(self))
 
-    def __copy__(self) -> "FrozenList":
+    def __copy__(self) -> FrozenList:
         return FrozenList(self)
 
-    def __deepcopy__(self, memo: Dict) -> "FrozenList":
+    def __deepcopy__(self, memo: Dict) -> FrozenList:
         return FrozenList(deepcopy(value, memo) for value in self)
 
     def append(self, x):

src/graphql/pyutils/path.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from typing import Any, List, NamedTuple, Optional, Union
 
 __all__ = ["Path"]
@@ -13,7 +15,7 @@ class Path(NamedTuple):
     typename: Optional[str]
     """name of the parent type to avoid path ambiguity"""
 
-    def add_key(self, key: Union[str, int], typename: Optional[str] = None) -> "Path":
+    def add_key(self, key: Union[str, int], typename: Optional[str] = None) -> Path:
         """Return a new Path containing the given key."""
         return Path(self, key, typename)
 

src/graphql/pyutils/simple_pub_sub.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from asyncio import Future, Queue, ensure_future, sleep
 from inspect import isawaitable
 from typing import Any, AsyncIterator, Callable, Optional, Set
@@ -31,7 +33,7 @@ def emit(self, event: Any) -> bool:
 
     def get_subscriber(
         self, transform: Optional[Callable] = None
-    ) -> "SimplePubSubIterator":
+    ) -> SimplePubSubIterator:
         return SimplePubSubIterator(self, transform)
 
 
@@ -44,7 +46,7 @@ def __init__(self, pubsub: SimplePubSub, transform: Optional[Callable]) -> None:
         self.listening = True
         pubsub.subscribers.add(self.push_value)
 
-    def __aiter__(self) -> "SimplePubSubIterator":
+    def __aiter__(self) -> SimplePubSubIterator:
         return self
 
     async def __anext__(self) -> Any:

src/graphql/type/definition.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from enum import Enum
 from typing import (
     Any,
@@ -280,7 +282,7 @@ def to_kwargs(self) -> GraphQLNamedTypeKwargs:
             extension_ast_nodes=self.extension_ast_nodes,
         )
 
-    def __copy__(self) -> "GraphQLNamedType":  # pragma: no cover
+    def __copy__(self) -> GraphQLNamedType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 
@@ -451,7 +453,7 @@ def to_kwargs(self) -> GraphQLScalarTypeKwargs:
             specified_by_url=self.specified_by_url,
         )
 
-    def __copy__(self) -> "GraphQLScalarType":  # pragma: no cover
+    def __copy__(self) -> GraphQLScalarType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 
@@ -469,7 +471,7 @@ def assert_scalar_type(type_: Any) -> GraphQLScalarType:
 
 
 class GraphQLFieldKwargs(TypedDict, total=False):
-    type_: "GraphQLOutputType"
+    type_: GraphQLOutputType
     args: Optional[GraphQLArgumentMap]
     resolve: Optional["GraphQLFieldResolver"]
     subscribe: Optional["GraphQLFieldResolver"]
@@ -482,7 +484,7 @@ class GraphQLFieldKwargs(TypedDict, total=False):
 class GraphQLField:
     """Definition of a GraphQL field"""
 
-    type: "GraphQLOutputType"
+    type: GraphQLOutputType
     args: GraphQLArgumentMap
     resolve: Optional["GraphQLFieldResolver"]
     subscribe: Optional["GraphQLFieldResolver"]
@@ -493,7 +495,7 @@ class GraphQLField:
 
     def __init__(
         self,
-        type_: "GraphQLOutputType",
+        type_: GraphQLOutputType,
         args: Optional[GraphQLArgumentMap] = None,
         resolve: Optional["GraphQLFieldResolver"] = None,
         subscribe: Optional["GraphQLFieldResolver"] = None,
@@ -577,7 +579,7 @@ def to_kwargs(self) -> GraphQLFieldKwargs:
             ast_node=self.ast_node,
         )
 
-    def __copy__(self) -> "GraphQLField":  # pragma: no cover
+    def __copy__(self) -> GraphQLField:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 
@@ -593,10 +595,10 @@ class GraphQLResolveInfo(NamedTuple):
 
     field_name: str
     field_nodes: List[FieldNode]
-    return_type: "GraphQLOutputType"
-    parent_type: "GraphQLObjectType"
+    return_type: GraphQLOutputType
+    parent_type: GraphQLObjectType
     path: Path
-    schema: "GraphQLSchema"
+    schema: GraphQLSchema
     fragments: Dict[str, FragmentDefinitionNode]
     root_value: Any
     operation: OperationDefinitionNode
@@ -628,7 +630,7 @@ class GraphQLResolveInfo(NamedTuple):
 
 
 class GraphQLArgumentKwargs(TypedDict, total=False):
-    type_: "GraphQLInputType"
+    type_: GraphQLInputType
     default_value: Any
     description: Optional[str]
     deprecation_reason: Optional[str]
@@ -640,7 +642,7 @@ class GraphQLArgumentKwargs(TypedDict, total=False):
 class GraphQLArgument:
     """Definition of a GraphQL argument"""
 
-    type: "GraphQLInputType"
+    type: GraphQLInputType
     default_value: Any
     description: Optional[str]
     deprecation_reason: Optional[str]
@@ -650,7 +652,7 @@ class GraphQLArgument:
 
     def __init__(
         self,
-        type_: "GraphQLInputType",
+        type_: GraphQLInputType,
         default_value: Any = Undefined,
         description: Optional[str] = None,
         deprecation_reason: Optional[str] = None,
@@ -706,7 +708,7 @@ def to_kwargs(self) -> GraphQLArgumentKwargs:
             ast_node=self.ast_node,
         )
 
-    def __copy__(self) -> "GraphQLArgument":  # pragma: no cover
+    def __copy__(self) -> GraphQLArgument:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 
@@ -798,7 +800,7 @@ def to_kwargs(self) -> GraphQLObjectTypeKwargs:
             is_type_of=self.is_type_of,
         )
 
-    def __copy__(self) -> "GraphQLObjectType":  # pragma: no cover
+    def __copy__(self) -> GraphQLObjectType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
     @cached_property
@@ -932,7 +934,7 @@ def to_kwargs(self) -> GraphQLInterfaceTypeKwargs:
             resolve_type=self.resolve_type,
         )
 
-    def __copy__(self) -> "GraphQLInterfaceType":  # pragma: no cover
+    def __copy__(self) -> GraphQLInterfaceType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
     @cached_property
@@ -1063,7 +1065,7 @@ def to_kwargs(self) -> GraphQLUnionTypeKwargs:
             super().to_kwargs(), types=self.types, resolve_type=self.resolve_type
         )
 
-    def __copy__(self) -> "GraphQLUnionType":  # pragma: no cover
+    def __copy__(self) -> GraphQLUnionType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
     @cached_property
@@ -1203,7 +1205,7 @@ def to_kwargs(self) -> GraphQLEnumTypeKwargs:
             super().to_kwargs(), values=self.values.copy()
         )
 
-    def __copy__(self) -> "GraphQLEnumType":  # pragma: no cover
+    def __copy__(self) -> GraphQLEnumType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
     @cached_property
@@ -1352,7 +1354,7 @@ def to_kwargs(self) -> GraphQLEnumValueKwargs:
             ast_node=self.ast_node,
         )
 
-    def __copy__(self) -> "GraphQLEnumValue":  # pragma: no cover
+    def __copy__(self) -> GraphQLEnumValue:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 
@@ -1446,7 +1448,7 @@ def to_kwargs(self) -> GraphQLInputObjectTypeKwargs:
             else self.out_type,
         )
 
-    def __copy__(self) -> "GraphQLInputObjectType":  # pragma: no cover
+    def __copy__(self) -> GraphQLInputObjectType:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
     @cached_property
@@ -1491,7 +1493,7 @@ def assert_input_object_type(type_: Any) -> GraphQLInputObjectType:
 
 
 class GraphQLInputFieldKwargs(TypedDict, total=False):
-    type_: "GraphQLInputType"
+    type_: GraphQLInputType
     default_value: Any
     description: Optional[str]
     deprecation_reason: Optional[str]
@@ -1503,7 +1505,7 @@ class GraphQLInputFieldKwargs(TypedDict, total=False):
 class GraphQLInputField:
     """Definition of a GraphQL input field"""
 
-    type: "GraphQLInputType"
+    type: GraphQLInputType
     default_value: Any
     description: Optional[str]
     deprecation_reason: Optional[str]
@@ -1513,7 +1515,7 @@ class GraphQLInputField:
 
     def __init__(
         self,
-        type_: "GraphQLInputType",
+        type_: GraphQLInputType,
         default_value: Any = Undefined,
         description: Optional[str] = None,
         deprecation_reason: Optional[str] = None,
@@ -1569,7 +1571,7 @@ def to_kwargs(self) -> GraphQLInputFieldKwargs:
             ast_node=self.ast_node,
         )
 
-    def __copy__(self) -> "GraphQLInputField":  # pragma: no cover
+    def __copy__(self) -> GraphQLInputField:  # pragma: no cover
         return self.__class__(**self.to_kwargs())
 
 

src/graphql/type/directives.py

@@ -1,3 +1,5 @@
+from __future__ import annotations  # Python < 3.10
+
 from typing import Any, Collection, Dict, Optional, Tuple, cast
 
 from ..language import ast, DirectiveLocation
@@ -143,7 +145,7 @@ def to_kwargs(self) -> GraphQLDirectiveKwargs:
             ast_node=self.ast_node,
         )
 
-    def __copy__(self) -> "GraphQLDirective":  # pragma: no cover
+    def __copy__(self) -> GraphQLDirective:  # pragma: no cover
         return self.__class__(**self.to_kwargs())