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

Add support to return a promise for connections.

Browse Source
This commit is contained in:
Markus Padourek 2016-10-26 10:50:27 +01:00 committed by GitHub
parent 0a80119f5e
commit ea69be6117
1 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
graphene/relay/connection.py
View File

@ -118,26 +118,29 @@ class IterableConnectionField(Field):
return connection_type return connection_type
@classmethod @classmethod
def connection_resolver(cls, resolver, connection, root, args, context, info): def connection_resolver(cls, resolver, connection_type, root, args, context, info):
resolved = resolver(root, args, context, info) p = Promise.resolve(resolver(root, args, context, info))
if isinstance(resolved, connection): def resolve_connection(resolved):
if isinstance(resolved, connection_type):
return resolved return resolved
assert isinstance(resolved, Iterable), ( assert isinstance(resolved, Iterable), (
'Resolved value from the connection field have to be iterable or instance of {}. ' 'Resolved value from the connection field have to be iterable or instance of {}. '
'Received "{}"' 'Received "{}"'
).format(connection, resolved) ).format(connection_type, resolved)
connection = connection_from_list( connection = connection_from_list(
resolved, resolved,
args, args,
connection_type=connection, connection_type=connection_type,
edge_type=connection.Edge, edge_type=connection_type.Edge,
pageinfo_type=PageInfo pageinfo_type=PageInfo
) )
connection.iterable = resolved connection.iterable = resolved
return connection return connection
return p.then(resolve_connection)
def get_resolver(self, parent_resolver): def get_resolver(self, parent_resolver):
resolver = super(IterableConnectionField, self).get_resolver(parent_resolver) resolver = super(IterableConnectionField, self).get_resolver(parent_resolver)
return partial(self.connection_resolver, resolver, self.type) return partial(self.connection_resolver, resolver, self.type)