fix: fix mypy issues

strawberry-graphql · Dec 26, 2024 · 275148b · 275148b
1 parent a63500a
commit 275148b
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 11 deletions.
diff --git a/strawberry/codegen/plugins/typescript.py b/strawberry/codegen/plugins/typescript.py
@@ -102,6 +102,7 @@ def _print_scalar_type(self, type_: GraphQLScalar) -> str:
         if type_.name in self.SCALARS_TO_TS_TYPE:
             return ""
 
+        assert type_.python_type is not None
         return f"type {type_.name} = {self.SCALARS_TO_TS_TYPE[type_.python_type]}"
 
     def _print_union_type(self, type_: GraphQLUnion) -> str:

diff --git a/strawberry/extensions/context.py b/strawberry/extensions/context.py
@@ -146,20 +146,22 @@ def from_callable(
         func: Callable[[SchemaExtension], AwaitableOrValue[Any]],
     ) -> WrappedHook:
         if iscoroutinefunction(func):
+            is_async = True
 
             @contextlib.asynccontextmanager
             async def iterator() -> AsyncIterator[None]:
                 await func(extension)
                 yield
 
-            return WrappedHook(extension=extension, hook=iterator, is_async=True)
+        else:
+            is_async = False
 
-        @contextlib.contextmanager
-        def iterator() -> Iterator[None]:
-            func(extension)
-            yield
+            @contextlib.contextmanager
+            def iterator() -> Iterator[None]:
+                func(extension)
+                yield
 
-        return WrappedHook(extension=extension, hook=iterator, is_async=False)
+        return WrappedHook(extension=extension, hook=iterator, is_async=is_async)
 
     def __enter__(self) -> None:
         self.exit_stack = contextlib.ExitStack()

diff --git a/strawberry/federation/schema.py b/strawberry/federation/schema.py
@@ -271,7 +271,7 @@ def _add_link_directives(
         link_directives: list[object] = [
             Link(
                 url=url,
-                import_=sorted(directives),
+                import_=sorted(directives),  # type: ignore[arg-type]
             )
             for url, directives in directive_by_url.items()
         ]

diff --git a/strawberry/litestar/controller.py b/strawberry/litestar/controller.py
@@ -237,7 +237,7 @@ class GraphQLController(
     ],
 ):
     path: str = ""
-    dependencies: ClassVar[Dependencies] = {
+    dependencies: ClassVar[Dependencies] = {  # type: ignore[misc]
         "custom_context": Provide(_none_custom_context_getter),
         "context": Provide(_context_getter_http),
         "context_ws": Provide(_context_getter_ws),
@@ -446,7 +446,7 @@ def make_graphql_controller(
 
     class _GraphQLController(GraphQLController):
         path: str = routes_path
-        dependencies: ClassVar[Dependencies] = {
+        dependencies: ClassVar[Dependencies] = {  # type: ignore[misc]
             "custom_context": Provide(custom_context_getter_),
             "context": Provide(_context_getter_http),
             "context_ws": Provide(_context_getter_ws),

diff --git a/strawberry/types/arguments.py b/strawberry/types/arguments.py
@@ -24,7 +24,7 @@
 from strawberry.types.enum import EnumDefinition
 from strawberry.types.lazy_type import LazyType, StrawberryLazyReference
 from strawberry.types.unset import UNSET as _deprecated_UNSET  # noqa: N811
-from strawberry.types.unset import _deprecated_is_unset  # type: ignore  # noqa: F401
+from strawberry.types.unset import _deprecated_is_unset  # noqa: F401
 
 if TYPE_CHECKING:
     from strawberry.schema.config import StrawberryConfig

diff --git a/strawberry/utils/graphql_lexer.py b/strawberry/utils/graphql_lexer.py
@@ -12,7 +12,7 @@ class GraphQLLexer(RegexLexer):
     filenames: ClassVar[list[str]] = ["*.graphql", "*.gql"]
     mimetypes: ClassVar[list[str]] = ["application/graphql"]
 
-    tokens: ClassVar[dict[str, tuple[str, Any]]] = {
+    tokens: ClassVar[dict[str, list[tuple[str, Any]]]] = {
         "root": [
             (r"#.*", token.Comment.Singline),
             (r"\.\.\.", token.Operator),