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

Accept source_loader from kwargs in DataLoaderField resolver

Browse Source
This commit is contained in:
Olivia Rodriguez Valdes 2019-04-10 14:16:08 -04:00
parent 581dbe7416
commit 9caa57a7ff
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
graphene_django/fields.py
View File

@ -188,7 +188,10 @@ class DataLoaderField(DjangoField):
# If no resolver is explicitly provided, use dataloader # If no resolver is explicitly provided, use dataloader
self.resolver = self.resolver or self.resolver_data_loader self.resolver = self.resolver or self.resolver_data_loader
def resolver_data_loader(self, root, info, *args): def resolver_data_loader(self, root, info, *args, **kwargs):
"""Resolve field through dataloader""" """Resolve field through dataloader"""
source_loader = getattr(root, self.source_loader) if root:
source_loader = getattr(root, self.source_loader)
else:
source_loader = kwargs.get(self.source_loader)
return self.data_loader.load(source_loader) return self.data_loader.load(source_loader)