Update mypy, use new semantic analyzer (#67)

Author: Cito

mypy.ini

@@ -3,5 +3,3 @@ python_version = 3.8
 check_untyped_defs = True
 warn_redundant_casts = True
 warn_unused_ignores = True
-# https://github.com/python/mypy/issues/7203
-new_semantic_analyzer = False

poetry.lock

@@ -40,7 +40,7 @@ pytest-describe = ">=0.12"
 pyyaml = "^5.1"
 black = ">=19.10b0"
 flake8 = "^3.7"
-mypy = ">=0.720,<0.730"
+mypy = ">=0.750,<0.760"
 codecov = "^2"
 sphinx = "^2.2"
 sphinx_rtd_theme = ">=0.4"

pyproject.toml

@@ -34,7 +34,8 @@
     FrozenList,
     Path,
 )
-from ..utilities import get_operation_root_type, type_from_ast
+from ..utilities.get_operation_root_type import get_operation_root_type
+from ..utilities.type_from_ast import type_from_ast
 from ..type import (
     GraphQLAbstractType,
     GraphQLField,

src/graphql/execution/execute.py

@@ -23,7 +23,9 @@
     is_input_type,
     is_non_null_type,
 )
-from ..utilities import coerce_input_value, type_from_ast, value_from_ast
+from ..utilities.coerce_input_value import coerce_input_value
+from ..utilities.type_from_ast import type_from_ast
+from ..utilities.value_from_ast import value_from_ast
 
 __all__ = ["get_variable_values", "get_argument_values", "get_directive_values"]
 

src/graphql/execution/values.py

@@ -239,7 +239,7 @@ def __init_subclass__(cls):
         keys: List[str] = []
         for base in cls.__bases__:
             # noinspection PyUnresolvedReferences
-            keys.extend(base.keys)
+            keys.extend(base.keys)  # type: ignore
         keys.extend(cls.__slots__)
         cls.keys = keys
 

src/graphql/language/ast.py

@@ -5,6 +5,7 @@
     Callable,
     List,
     NamedTuple,
+    Optional,
     Sequence,
     Tuple,
     Union,
@@ -157,8 +158,11 @@ def __init_subclass__(cls):
         for attr, val in cls.__dict__.items():
             if attr.startswith("_"):
                 continue
-            attr = attr.split("_", 1)
-            attr, kind = attr if len(attr) > 1 else (attr[0], None)
+            attr_kind = attr.split("_", 1)
+            if len(attr_kind) < 2:
+                kind: Optional[str] = None
+            else:
+                attr, kind = attr_kind
             if attr in ("enter", "leave"):
                 if kind:
                     name = snake_to_camel(kind) + "Node"

src/graphql/language/visitor.py

@@ -1,7 +1,7 @@
-from asyncio import Event, ensure_future, wait
+from asyncio import Event, ensure_future, Future, wait
 from concurrent.futures import FIRST_COMPLETED
 from inspect import isasyncgen, isawaitable
-from typing import AsyncIterable, Callable
+from typing import AsyncIterable, Callable, Set
 
 __all__ = ["MapAsyncIterator"]
 
@@ -42,7 +42,9 @@ async def __anext__(self):
             aclose = ensure_future(self._close_event.wait())
             anext = ensure_future(self.iterator.__anext__())
 
-            done, pending = await wait([aclose, anext], return_when=FIRST_COMPLETED)
+            pending: Set[Future] = (
+                await wait([aclose, anext], return_when=FIRST_COMPLETED)
+            )[1]
             for task in pending:
                 task.cancel()
 

src/graphql/subscription/map_async_iterator.py

@@ -62,7 +62,7 @@ async def subscribe(
     if isinstance(result_or_stream, ExecutionResult):
         return result_or_stream
 
-    async def map_source_to_response(payload):
+    async def map_source_to_response(payload) -> ExecutionResult:
         """Map source to response.
 
         For each payload yielded from a subscription, map it over the normal GraphQL
@@ -81,7 +81,7 @@ async def map_source_to_response(payload):
             operation_name,
             field_resolver,
         )
-        return await result if isawaitable(result) else result
+        return await result if isawaitable(result) else result  # type: ignore
 
     return MapAsyncIterator(result_or_stream, map_source_to_response)
 

src/graphql/subscription/subscribe.py

@@ -361,11 +361,11 @@ def __init__(
                 " as a sequence of ScalarTypeExtensionNode instances."
             )
         if serialize is not None:
-            self.serialize = serialize
+            self.serialize = serialize  # type: ignore
         if parse_value is not None:
-            self.parse_value = parse_value
+            self.parse_value = parse_value  # type: ignore
         if parse_literal is not None:
-            self.parse_literal = parse_literal
+            self.parse_literal = parse_literal  # type: ignore
 
     def __repr__(self):
         return f"<{self.__class__.__name__} {self.name!r}>"
@@ -391,9 +391,7 @@ def parse_value(value: Any) -> Any:
         """
         return value
 
-    def parse_literal(  # type: ignore
-        self, node: ValueNode, _variables: Dict[str, Any] = None
-    ) -> Any:
+    def parse_literal(self, node: ValueNode, _variables: Dict[str, Any] = None) -> Any:
         """Parses an externally provided literal value to use as an input.
 
         This default method uses the parse_value method and should be replaced
@@ -1247,7 +1245,7 @@ def __init__(
             )
         self._fields = fields
         if out_type is not None:
-            self.out_type = out_type
+            self.out_type = out_type  # type: ignore
 
     @staticmethod
     def out_type(value: Dict[str, Any]) -> Any: