44from graphql .type .definition import GraphQLObjectType , GraphQLField , GraphQLScalarType
55
66
7- class TypeAdaptor (object ):
7+ class TypeAdapter (object ):
88 """Substitute custom scalars in a GQL response with their decoded counterparts.
99
1010 GQL custom scalar types are defined on the GQL schema and are used to represent
@@ -16,20 +16,20 @@ class TypeAdaptor(object):
1616 the `_traverse()` function).
1717
1818 Each time we find a field which is a custom scalar (it's type name appears
19- as a key in self.custom_scalars ), we replace the value of that field with the
19+ as a key in self.custom_types ), we replace the value of that field with the
2020 decoded value. All of this logic happens in `_substitute()`.
2121
2222 Public Interface:
2323 apply(): pass in a GQL response to replace all instances of custom
2424 scalar strings with their deserialized representation."""
2525
26- def __init__ (self , schema : GraphQLSchema , custom_scalars : Dict [str , Any ] = {}) -> None :
26+ def __init__ (self , schema : GraphQLSchema , custom_types : Dict [str , Any ] = {}) -> None :
2727 """ schema: a graphQL schema in the GraphQLSchema format
28- custom_scalars : a Dict[str, Any],
28+ custom_types : a Dict[str, Any],
2929 where str is the name of the custom scalar type, and
30- Any is a class which has a `parse_value()` function"""
30+ Any is a class which has a `parse_value(str )` function"""
3131 self .schema = schema
32- self .custom_scalars = custom_scalars
32+ self .custom_types = custom_types
3333
3434 def _follow_type_chain (self , node : Any ) -> Any :
3535 """ Get the type of the schema node in question.
@@ -61,46 +61,49 @@ def _lookup_scalar_type(self, keys: List[str]) -> Optional[str]:
6161 If keys (e.g. ['film', 'release_date']) points to a scalar type, then
6262 this function returns the name of that type. (e.g. 'DateTime')
6363
64- If it is not a scalar type (e..g a GraphQLObject or list ), then this
64+ If it is not a scalar type (e..g a GraphQLObject), then this
6565 function returns None.
6666
6767 `keys` is a breadcrumb trail telling us where to look in the GraphQL schema.
6868 By default the root level is `schema.query`, if that fails, then we check
6969 `schema.mutation`."""
7070
71- def iterate (node : Any , lookup : List [str ]):
72- lookup = lookup .copy ()
71+ def traverse_schema (node : Any , lookup : List [str ]):
7372 if not lookup :
7473 return self ._get_scalar_type_name (node )
7574
7675 final_node = self ._follow_type_chain (node )
77- return iterate (final_node .fields [lookup .pop (0 )], lookup )
76+ return traverse_schema (final_node .fields [lookup [0 ]], lookup [1 :])
77+
78+ if keys [0 ] in self .schema .get_query_type ().fields :
79+ schema_root = self .schema .get_query_type ()
80+ elif keys [0 ] in self .schema .get_mutation_type ().fields :
81+ schema_root = self .schema .get_mutation_type ()
82+ else :
83+ return None
7884
7985 try :
80- return iterate ( self . schema . get_query_type () , keys )
86+ return traverse_schema ( schema_root , keys )
8187 except (KeyError , AttributeError ):
82- try :
83- return iterate (self .schema .get_mutation_type (), keys )
84- except (KeyError , AttributeError ):
85- return None
88+ return None
8689
87- def _substitute (self , keys : List [str ], value : Any ) -> Any :
90+ def _get_decoded_scalar_type (self , keys : List [str ], value : Any ) -> Any :
8891 """Get the decoded value of the type identified by `keys`.
8992
9093 If the type is not a custom scalar, then return the original value.
9194
9295 If it is a custom scalar, return the deserialized value, as
9396 output by `<CustomScalarType>.parse_value()`"""
9497 scalar_type = self ._lookup_scalar_type (keys )
95- if scalar_type and scalar_type in self .custom_scalars :
96- return self .custom_scalars [scalar_type ].parse_value (value )
98+ if scalar_type and scalar_type in self .custom_types :
99+ return self .custom_types [scalar_type ].parse_value (value )
97100 return value
98101
99- def _traverse (self , response : Dict [str , Any ], substitute : Callable ) -> Dict [str , Any ]:
102+ def convert_scalars (self , response : Dict [str , Any ]) -> Dict [str , Any ]:
100103 """Recursively traverse the GQL response
101104
102- Recursively traverses the GQL response and calls the `substitute`
103- function on all leaf nodes. The function is called with 2 arguments:
105+ Recursively traverses the GQL response and calls _get_decoded_scalar_type()
106+ for all leaf nodes. The function is called with 2 arguments:
104107 keys: List[str] is a breadcrumb trail telling us where we are in the
105108 response, and therefore, where to look in the GQL Schema.
106109 value: Any is the value at that node in the response
@@ -109,15 +112,9 @@ def _traverse(self, response: Dict[str, Any], substitute: Callable) -> Dict[str,
109112 modified."""
110113 def iterate (node : Any , keys : List [str ] = []):
111114 if isinstance (node , dict ):
112- result = {}
113- for _key , value in node .items ():
114- result [_key ] = iterate (value , keys + [_key ])
115- return result
115+ return {_key : iterate (value , keys + [_key ]) for _key , value in node .items ()}
116116 elif isinstance (node , list ):
117117 return [(iterate (item , keys )) for item in node ]
118118 else :
119- return substitute (keys , node )
119+ return self . _get_decoded_scalar_type (keys , node )
120120 return iterate (response )
121-
122- def apply (self , response : Dict [str , Any ]) -> Dict [str , Any ]:
123- return self ._traverse (response , self ._substitute )
0 commit comments