Fix typing

Author: Cito

graphql/utilities/extend_schema.py

@@ -17,14 +17,10 @@
     GraphQLArgumentMap,
     GraphQLDirective,
     GraphQLEnumType,
-    GraphQLEnumValue,
-    GraphQLEnumValueMap,
     GraphQLField,
     GraphQLFieldMap,
     GraphQLInputField,
-    GraphQLInputFieldMap,
     GraphQLInputObjectType,
-    GraphQLInputType,
     GraphQLInterfaceType,
     GraphQLList,
     GraphQLNamedType,

graphql/utilities/lexicographic_sort_schema.py

@@ -100,21 +100,25 @@ def sort_named_type_impl(type_: GraphQLNamedType) -> GraphQLNamedType:
             return type_
         elif is_object_type(type_):
             kwargs = type_.to_kwargs()
+            object_type = cast(GraphQLObjectType, type_)
             kwargs.update(
-                interfaces=lambda: sort_types(type_.interfaces),
-                fields=lambda: sort_fields(type_.fields),
+                interfaces=lambda: sort_types(object_type.interfaces),
+                fields=lambda: sort_fields(object_type.fields),
             )
             return GraphQLObjectType(**kwargs)
         elif is_interface_type(type_):
             kwargs = type_.to_kwargs()
-            kwargs.update(fields=lambda: sort_fields(type_.fields))
+            interface_type = cast(GraphQLInterfaceType, type_)
+            kwargs.update(fields=lambda: sort_fields(interface_type.fields))
             return GraphQLInterfaceType(**kwargs)
         elif is_union_type(type_):
             kwargs = type_.to_kwargs()
-            kwargs.update(types=lambda: sort_types(type_.types))
+            union_type = cast(GraphQLUnionType, type_)
+            kwargs.update(types=lambda: sort_types(union_type.types))
             return GraphQLUnionType(**kwargs)
         elif is_enum_type(type_):
             kwargs = type_.to_kwargs()
+            enum_type = cast(GraphQLEnumType, type_)
             kwargs.update(
                 values={
                     name: GraphQLEnumValue(
@@ -123,13 +127,14 @@ def sort_named_type_impl(type_: GraphQLNamedType) -> GraphQLNamedType:
                         deprecation_reason=val.deprecation_reason,
                         ast_node=val.ast_node,
                     )
-                    for name, val in sorted(type_.values.items())
+                    for name, val in sorted(enum_type.values.items())
                 }
             )
             return GraphQLEnumType(**kwargs)
         elif is_input_object_type(type_):
             kwargs = type_.to_kwargs()
-            kwargs.update(fields=sort_input_fields(type_.fields))
+            input_object_type = cast(GraphQLInputObjectType, type_)
+            kwargs.update(fields=sort_input_fields(input_object_type.fields))
             return GraphQLInputObjectType(**kwargs)
         raise TypeError(f"Unknown type: '{type_}'")