graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-03-05 20:35:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'patch-11'

Browse Source
This commit is contained in:
Syrus Akbary 2016-10-27 02:42:05 +02:00
commit 0b9aa7cbeb
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
graphene/relay/connection.py
View File

@ -5,6 +5,7 @@ from functools import partial
import six
from graphql_relay import connection_from_list
from promise import Promise
from ..types import (AbstractType, Boolean, Enum, Int, Interface, List, NonNull, Scalar, String,
Union)
@ -118,26 +119,34 @@ class IterableConnectionField(Field):
return connection_type
@classmethod
def connection_resolver(cls, resolver, connection, root, args, context, info):
resolved = resolver(root, args, context, info)
if isinstance(resolved, connection):
def resolve_connection(cls, connection_type, args, resolved):
if isinstance(resolved, connection_type):
return resolved
assert isinstance(resolved, Iterable), (
'Resolved value from the connection field have to be iterable or instance of {}. '
'Received "{}"'
).format(connection, resolved)
).format(connection_type, resolved)
connection = connection_from_list(
resolved,
args,
connection_type=connection,
edge_type=connection.Edge,
connection_type=connection_type,
edge_type=connection_type.Edge,
pageinfo_type=PageInfo
)
connection.iterable = resolved
return connection
@classmethod
def connection_resolver(cls, resolver, connection_type, root, args, context, info):
resolved = resolver(root, args, context, info)
on_resolve = partial(cls.resolve_connection, connection_type, args)
if isinstance(resolved, Promise):
return resolved.then(on_resolve)
return on_resolve(resolved)
def get_resolver(self, parent_resolver):
resolver = super(IterableConnectionField, self).get_resolver(parent_resolver)
return partial(self.connection_resolver, resolver, self.type)