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

accept connections for DjangoFilterConnectionField

Browse Source
This commit is contained in:
Nick Hudkins 2017-01-27 10:47:56 -05:00
parent 5d6c7f2dc3
commit 15664bdc0b
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
graphene_django/filter/fields.py
View File

@ -1,6 +1,7 @@
from functools import partial
from ..fields import DjangoConnectionField
from graphene.relay import is_node
from .utils import get_filtering_args_from_filterset, get_filterset_class
@ -9,9 +10,18 @@ class DjangoFilterConnectionField(DjangoConnectionField):
def __init__(self, type, fields=None, extra_filter_meta=None,
filterset_class=None, *args, **kwargs):
self.fields = fields or type._meta.filter_fields
meta = dict(model=type._meta.model,
fields=self.fields)
if is_node(type):
_fields = type._meta.filter_fields
_model = type._meta.model
else:
# ConnectionFields can also be passed Connections,
# in which case, we need to use the Node of the connection
# to get our relevant args.
_fields = type._meta.node._meta.filter_fields
_model = type._meta.node._meta.model
self.fields = fields or _fields
meta = dict(model=_model, fields=self.fields)
if extra_filter_meta:
meta.update(extra_filter_meta)
self.filterset_class = get_filterset_class(filterset_class, **meta)