graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2024-11-27 03:53:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed circular union. Fixed #439

Browse Source
This commit is contained in:
Syrus Akbary 2017-03-24 22:40:16 -07:00
parent a837fb6836
commit d4d8a76a09
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
graphene/types/typemap.py
View File

@ -81,16 +81,18 @@ class TypeMap(GraphQLTypeMap):
if issubclass(type, ObjectType): if issubclass(type, ObjectType):
internal_type = self.construct_objecttype(map, type) internal_type = self.construct_objecttype(map, type)
if issubclass(type, InputObjectType): elif issubclass(type, InputObjectType):
internal_type = self.construct_inputobjecttype(map, type) internal_type = self.construct_inputobjecttype(map, type)
if issubclass(type, Interface): elif issubclass(type, Interface):
internal_type = self.construct_interface(map, type) internal_type = self.construct_interface(map, type)
if issubclass(type, Scalar): elif issubclass(type, Scalar):
internal_type = self.construct_scalar(map, type) internal_type = self.construct_scalar(map, type)
if issubclass(type, Enum): elif issubclass(type, Enum):
internal_type = self.construct_enum(map, type) internal_type = self.construct_enum(map, type)
if issubclass(type, Union): elif issubclass(type, Union):
internal_type = self.construct_union(map, type) internal_type = self.construct_union(map, type)
else:
raise Exception("Expected Graphene type, but received: {}.".format(type))
return GraphQLTypeMap.reducer(map, internal_type) return GraphQLTypeMap.reducer(map, internal_type)
@ -145,8 +147,10 @@ class TypeMap(GraphQLTypeMap):
def interfaces(): def interfaces():
interfaces = [] interfaces = []
for interface in type._meta.interfaces: for interface in type._meta.interfaces:
i = self.construct_interface(map, interface) self.graphene_reducer(map, interface)
interfaces.append(i) internal_type = map[interface._meta.name]
assert internal_type.graphene_type == interface
interfaces.append(internal_type)
return interfaces return interfaces
return GrapheneObjectType( return GrapheneObjectType(
@ -190,10 +194,16 @@ class TypeMap(GraphQLTypeMap):
_resolve_type = None _resolve_type = None
if type.resolve_type: if type.resolve_type:
_resolve_type = partial(resolve_type, type.resolve_type, map, type._meta.name) _resolve_type = partial(resolve_type, type.resolve_type, map, type._meta.name)
types = []
for i in type._meta.types: def types():
internal_type = self.construct_objecttype(map, i) union_types = []
types.append(internal_type) for objecttype in type._meta.types:
self.graphene_reducer(map, objecttype)
internal_type = map[objecttype._meta.name]
assert internal_type.graphene_type == objecttype
union_types.append(internal_type)
return union_types
return GrapheneUnionType( return GrapheneUnionType(
graphene_type=type, graphene_type=type,
name=type._meta.name, name=type._meta.name,