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

Add load-many support

Browse Source
This commit is contained in:
Olivia Rodriguez Valdes 2019-04-11 09:03:17 -04:00
parent 9caa57a7ff
commit 9128889353
2 changed files with 44 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
graphene_django/fields.py
View File

@ -171,17 +171,19 @@ class DjangoField(Field):
class DataLoaderField(DjangoField): class DataLoaderField(DjangoField):
"""Class to manage access to dataloader when resolve the field""" """Class to manage access to data-loader when resolve the field"""
def __init__(self, data_loader, source_loader, type, *args, **kwargs): def __init__(self, type, data_loader, source_loader, load_many=False, *args, **kwargs):
""" """
Initialization of dataloader to resolve field Initialization of data-loader to resolve field
:param data_loader: dataloader to resolve field :param data_loader: data-loader to resolve field
:param source_loader: field to obtain the key for dataloading :param source_loader: field to obtain the key for data-loading
:param load_many: Whether the resolver should try tu obtain one element or multiple elements
:param kwargs: Extra arguments :param kwargs: Extra arguments
""" """
self.data_loader = data_loader self.data_loader = data_loader
self.source_loader = source_loader self.source_loader = source_loader
self.load_many = load_many
super(DataLoaderField, self).__init__(type, *args, **kwargs) super(DataLoaderField, self).__init__(type, *args, **kwargs)
@ -191,7 +193,10 @@ class DataLoaderField(DjangoField):
def resolver_data_loader(self, root, info, *args, **kwargs): def resolver_data_loader(self, root, info, *args, **kwargs):
"""Resolve field through dataloader""" """Resolve field through dataloader"""
if root: if root:
source_loader = getattr(root, self.source_loader) source_loader = reduce(lambda x, y: getattr(x, y), self.source_loader.split('.'), root)
else: else:
source_loader = kwargs.get(self.source_loader) source_loader = kwargs.get(self.source_loader)
if self.load_many:
return self.data_loader.load_many(source_loader)
return self.data_loader.load(source_loader) return self.data_loader.load(source_loader)

33
graphene_django/tests/test_fields.py
View File

@ -9,6 +9,11 @@ from promise import Promise
class MyInstance(object): class MyInstance(object):
value = "value" value = "value"
key = 1 key = 1
keys = [1, 2, 3]
class InnerClass(object):
key = 2
keys = [4, 5, 6]
def resolver(self): def resolver(self):
return "resolver method" return "resolver method"
@ -62,6 +67,34 @@ class DataLoaderFieldTests(TestCase):
self.assertEqual(resolver(instance, None).get(), instance.key) self.assertEqual(resolver(instance, None).get(), instance.key)
def test_dataloaderfield_many(self):
MyType = object()
data_loader_field = DataLoaderField(data_loader=data_loader, source_loader='keys', type=MyType, load_many=True)
resolver = data_loader_field.get_resolver(None)
instance = MyInstance()
self.assertEqual(resolver(instance, None).get(), instance.keys)
def test_dataloaderfield_inner_prop(self):
MyType = object()
data_loader_field = DataLoaderField(data_loader=data_loader, source_loader='InnerClass.key', type=MyType)
resolver = data_loader_field.get_resolver(None)
instance = MyInstance()
self.assertEqual(resolver(instance, None).get(), instance.InnerClass.key)
def test_dataloaderfield_many_inner_prop(self):
MyType = object()
data_loader_field = DataLoaderField(data_loader=data_loader, source_loader='InnerClass.keys', type=MyType,
load_many=True)
resolver = data_loader_field.get_resolver(None)
instance = MyInstance()
self.assertEqual(resolver(instance, None).get(), instance.InnerClass.keys)
def test_dataloaderfield_permissions(self): def test_dataloaderfield_permissions(self):
MyType = object() MyType = object()
data_loader_field = DataLoaderField(data_loader=data_loader, source_loader='key', type=MyType, data_loader_field = DataLoaderField(data_loader=data_loader, source_loader='key', type=MyType,