graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-07-07 13:54:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

First phase of connection resolver

Browse Source
This commit is contained in:
Syrus Akbary 2016-11-02 21:25:17 -07:00
parent adfbffb267
commit bd07303971
1 changed files with 28 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
graphene/relay/connection.py
View File

@ -90,17 +90,39 @@ class ConnectionMeta(ObjectTypeMeta):
class Connection(six.with_metaclass(ConnectionMeta, ObjectType)): class Connection(six.with_metaclass(ConnectionMeta, ObjectType)):
pass
@classmethod
def Field(cls, *args, **kwargs): # noqa: N802
return ConnectionField(cls, *args, **kwargs)
@classmethod
def connection_resolver(cls, resolved, args, context, info):
if isinstance(resolved, cls):
return resolved
assert isinstance(resolved, Iterable), (
'Resolved value from the connection field have to be iterable or instance of {}. '
'Received "{}"'
).format(cls, resolved)
connection = connection_from_list(
resolved,
args,
connection_type=cls,
edge_type=cls.Edge,
pageinfo_type=PageInfo
)
connection.iterable = resolved
return connection
class IterableConnectionField(Field): class ConnectionField(Field):
def __init__(self, type, *args, **kwargs): def __init__(self, type, *args, **kwargs):
kwargs.setdefault('before', String()) kwargs.setdefault('before', String())
kwargs.setdefault('after', String()) kwargs.setdefault('after', String())
kwargs.setdefault('first', Int()) kwargs.setdefault('first', Int())
kwargs.setdefault('last', Int()) kwargs.setdefault('last', Int())
super(IterableConnectionField, self).__init__( super(ConnectionField, self).__init__(
type, type,
*args, *args,
**kwargs **kwargs
@ -108,7 +130,7 @@ class IterableConnectionField(Field):
@property @property
def type(self): def type(self):
type = super(IterableConnectionField, self).type type = super(ConnectionField, self).type
if is_node(type): if is_node(type):
connection_type = type.Connection connection_type = type.Connection
else: else:
@ -118,37 +140,16 @@ class IterableConnectionField(Field):
).format(str(self), connection_type) ).format(str(self), connection_type)
return connection_type return connection_type
@classmethod
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_type, resolved)
connection = connection_from_list(
resolved,
args,
connection_type=connection_type,
edge_type=connection_type.Edge,
pageinfo_type=PageInfo
)
connection.iterable = resolved
return connection
@classmethod @classmethod
def connection_resolver(cls, resolver, connection_type, root, args, context, info): def connection_resolver(cls, resolver, connection_type, root, args, context, info):
resolved = resolver(root, args, context, info) resolved = resolver(root, args, context, info)
on_resolve = partial(cls.resolve_connection, connection_type, args) on_resolve = partial(connection_type.connection_resolver, args=args, context=context, info=info)
if isinstance(resolved, Promise): if isinstance(resolved, Promise):
return resolved.then(on_resolve) return resolved.then(on_resolve)
return on_resolve(resolved) return on_resolve(resolved)
def get_resolver(self, parent_resolver): def get_resolver(self, parent_resolver):
resolver = super(IterableConnectionField, self).get_resolver(parent_resolver) resolver = super(ConnectionField, self).get_resolver(parent_resolver)
return partial(self.connection_resolver, resolver, self.type) return partial(self.connection_resolver, resolver, self.type)
ConnectionField = IterableConnectionField