graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2024-09-22 03:49:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added context to connectionfield resolver

Browse Source
This commit is contained in:
Markus Padourek 2016-05-20 13:59:39 +01:00
parent b431bfe477
commit c4f29f050b
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
graphene/relay/fields.py
View File

@ -5,7 +5,7 @@ from graphql_relay.node.node import from_global_id
from ..core.fields import Field
from ..core.types.definitions import NonNull
from ..core.types.scalars import ID, Int, String
from ..utils import with_context
from ..utils.wrap_resolver_function import has_context, with_context
class ConnectionField(Field):
@ -26,16 +26,23 @@ class ConnectionField(Field):
self.connection_type = connection_type
self.edge_type = edge_type
def resolver(self, instance, args, info):
@with_context
def resolver(self, instance, args, context, info):
schema = info.schema.graphene_schema
connection_type = self.get_type(schema)
resolver = super(ConnectionField, self).resolver
if has_context(resolver):
resolved = super(ConnectionField, self).resolver(instance, args, context, info)
else:
resolved = super(ConnectionField, self).resolver(instance, args, info)
if isinstance(resolved, connection_type):
return resolved
return self.from_list(connection_type, resolved, args, info)
return self.from_list(connection_type, resolved, args, context, info)
def from_list(self, connection_type, resolved, args, info):
return connection_type.from_list(resolved, args, info)
return connection_type.from_list(resolved, args, context, info)
def get_connection_type(self, node):
connection_type = self.connection_type or node.get_connection_type()