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

Merge pull request #99 from nickhudkins/feature/filter-connection-field-connections

Browse Source 
accept connections for DjangoFilterConnectionField
This commit is contained in:
Syrus Akbary 2017-02-17 10:36:32 -08:00 committed by GitHub
commit a599e4cb3e
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)