add operation_name to parse_results

Author: leszekhanusz

gql/client.py

@@ -350,7 +350,12 @@ def _execute(
         # Unserialize the result if requested
         if self.client.schema:
             if parse_result or (parse_result is None and self.client.parse_results):
-                result.data = parse_result_fn(self.client.schema, document, result.data)
+                result.data = parse_result_fn(
+                    self.client.schema,
+                    document,
+                    result.data,
+                    operation_name=operation_name,
+                )
 
         return result
 
@@ -495,7 +500,10 @@ async def _subscribe(
                         parse_result is None and self.client.parse_results
                     ):
                         result.data = parse_result_fn(
-                            self.client.schema, document, result.data
+                            self.client.schema,
+                            document,
+                            result.data,
+                            operation_name=operation_name,
                         )
 
                 yield result
@@ -611,7 +619,12 @@ async def _execute(
         # Unserialize the result if requested
         if self.client.schema:
             if parse_result or (parse_result is None and self.client.parse_results):
-                result.data = parse_result_fn(self.client.schema, document, result.data)
+                result.data = parse_result_fn(
+                    self.client.schema,
+                    document,
+                    result.data,
+                    operation_name=operation_name,
+                )
 
         return result
 

gql/utilities/parse_result.py

@@ -16,6 +16,7 @@
     GraphQLSchema,
     GraphQLType,
     InlineFragmentNode,
+    NameNode,
     Node,
     OperationDefinitionNode,
     SelectionSetNode,
@@ -71,6 +72,7 @@ def __init__(
         type_info: TypeInfo,
         visit_fragment: bool = False,
         inside_list_level: int = 0,
+        operation_name: Optional[str] = None,
     ):
         """Recursive Implementation of a Visitor class to parse results
         correspondind to a schema and a document.
@@ -96,6 +98,7 @@ def __init__(
         self.type_info: TypeInfo = type_info
         self.visit_fragment: bool = visit_fragment
         self.inside_list_level = inside_list_level
+        self.operation_name = operation_name
 
         self.result_stack: List[Any] = []
 
@@ -111,6 +114,22 @@ def leave_document(node: DocumentNode, *_args: Any) -> Dict[str, Any]:
         results = cast(List[Dict[str, Any]], node.definitions)
         return {k: v for result in results for k, v in result.items()}
 
+    def enter_operation_definition(
+        self, node: OperationDefinitionNode, *_args: Any
+    ) -> Union[None, VisitorActionEnum]:
+
+        if self.operation_name is not None:
+            if not hasattr(node.name, "value"):
+                return REMOVE  # pragma: no cover
+
+            node.name = cast(NameNode, node.name)
+
+            if node.name.value != self.operation_name:
+                log.debug(f"SKIPPING operation {node.name.value}")
+                return REMOVE
+
+        return IDLE
+
     @staticmethod
     def leave_operation_definition(
         node: OperationDefinitionNode, *_args: Any
@@ -374,6 +393,7 @@ def parse_result_recursive(
     initial_type: Optional[GraphQLType] = None,
     inside_list_level: int = 0,
     visit_fragment: bool = False,
+    operation_name: Optional[str] = None,
 ) -> Any:
 
     if result is None:
@@ -393,6 +413,7 @@ def parse_result_recursive(
                 type_info=type_info,
                 inside_list_level=inside_list_level,
                 visit_fragment=visit_fragment,
+                operation_name=operation_name,
             ),
         ),
         visitor_keys=RESULT_DOCUMENT_KEYS,
@@ -402,13 +423,17 @@ def parse_result_recursive(
 
 
 def parse_result(
-    schema: GraphQLSchema, document: DocumentNode, result: Optional[Dict[str, Any]],
+    schema: GraphQLSchema,
+    document: DocumentNode,
+    result: Optional[Dict[str, Any]],
+    operation_name: Optional[str] = None,
 ) -> Optional[Dict[str, Any]]:
     """Unserialize a result received from a GraphQL backend.
 
     :param schema: the GraphQL schema
     :param document: the document representing the query sent to the backend
     :param result: the serialized result received from the backend
+    :param operation_name: the optional operation name
 
     :returns: a parsed result with scalars and enums parsed depending on
               their definition in the schema.
@@ -423,4 +448,6 @@ def parse_result(
     will be parsed with the parse_value method of the custom scalar or enum
     definition in the schema."""
 
-    return parse_result_recursive(schema, document, document, result)
+    return parse_result_recursive(
+        schema, document, document, result, operation_name=operation_name
+    )

tests/custom_scalars/test_enum_colors.py

@@ -283,3 +283,43 @@ def test_update_schema_enum_errors():
         update_schema_enum(schema, "Color", {"RED": Color.RED})
 
     assert 'Enum key "GREEN" not found in provided values!' in str(exc_info)
+
+
+def test_parse_results_with_operation_type():
+
+    client = Client(schema=schema, parse_results=True)
+
+    query = gql(
+        """
+        query GetAll {
+            all
+        }
+        query GetOppositeColor($color: Color) {
+            opposite(color:$color)
+        }
+        query GetOppositeColor2($color: Color) {
+            other_opposite:opposite(color:$color)
+        }
+        query GetOppositeColor3 {
+            opposite(color: YELLOW)
+        }
+        query GetListOfListOfList {
+            list_of_list_of_list
+        }
+        """
+    )
+
+    variable_values = {
+        "color": "RED",
+    }
+
+    result = client.execute(
+        query, variable_values=variable_values, operation_name="GetOppositeColor"
+    )
+
+    print(result)
+
+    opposite_color = result["opposite"]
+
+    assert isinstance(opposite_color, Color)
+    assert opposite_color == CYAN