From 9caa57a7ff1d4852dff640b07c3b1a9772e55e74 Mon Sep 17 00:00:00 2001 From: Olivia Rodriguez Valdes Date: Wed, 10 Apr 2019 14:16:08 -0400 Subject: [PATCH] Accept source_loader from kwargs in DataLoaderField resolver --- graphene_django/fields.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/graphene_django/fields.py b/graphene_django/fields.py index 5de947f..2de0659 100644 --- a/graphene_django/fields.py +++ b/graphene_django/fields.py @@ -188,7 +188,10 @@ class DataLoaderField(DjangoField): # If no resolver is explicitly provided, use dataloader 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""" - 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)