This commit is contained in:
MardanovTimur 2019-02-24 02:45:47 +03:00
parent 0616e04513
commit 85d45414f9

View File

@ -27,6 +27,7 @@ class DjangoFilterConnectionField(DjangoConnectionField):
self._extra_filter_meta = extra_filter_meta self._extra_filter_meta = extra_filter_meta
self._base_args = None self._base_args = None
self._post_resolver = post_resolver self._post_resolver = post_resolver
self._custom_resolver = kwargs.pop('custom_resolver', False)
super(DjangoFilterConnectionField, self).__init__(type, *args, **kwargs) super(DjangoFilterConnectionField, self).__init__(type, *args, **kwargs)
@property @property
@ -94,6 +95,7 @@ class DjangoFilterConnectionField(DjangoConnectionField):
self.filterset_class, self.filterset_class,
self.filtering_args, self.filtering_args,
self._post_resolver, self._post_resolver,
self._custom_resolver,
) )
@classmethod @classmethod
@ -106,6 +108,7 @@ class DjangoFilterConnectionField(DjangoConnectionField):
filterset_class, filterset_class,
filtering_args, filtering_args,
post_resolver, post_resolver,
custom_resolver,
root, root,
info, info,
**args): **args):
@ -161,6 +164,9 @@ class DjangoFilterConnectionField(DjangoConnectionField):
return instances return instances
return qs return qs
if custom_resolver:
new_resolver = resolver
return DjangoConnectionField.connection_resolver( return DjangoConnectionField.connection_resolver(
new_resolver, new_resolver,
connection, connection,