diff --git a/graphene_django/fields.py b/graphene_django/fields.py index 1ecce45..308e85e 100644 --- a/graphene_django/fields.py +++ b/graphene_django/fields.py @@ -3,6 +3,7 @@ from functools import partial from django.db.models.query import QuerySet from promise import Promise +from neomodel import NodeSet from graphene.types import Field, List from graphene.relay import ConnectionField, PageInfo @@ -80,7 +81,7 @@ class DjangoConnectionField(ConnectionField): if iterable is None: iterable = default_manager iterable = maybe_queryset(iterable) - if isinstance(iterable, QuerySet): + if isinstance(iterable, NodeSet): if iterable is not default_manager: default_queryset = maybe_queryset(default_manager) iterable = cls.merge_querysets(default_queryset, iterable) diff --git a/graphene_django/types.py b/graphene_django/types.py index e306fd1..94a8344 100644 --- a/graphene_django/types.py +++ b/graphene_django/types.py @@ -54,6 +54,7 @@ class DjangoObjectType(ObjectType): only_fields=(), exclude_fields=(), filter_fields=None, + neomodel_filter_fields=None, connection=None, connection_class=None, use_connection=None, @@ -106,6 +107,7 @@ class DjangoObjectType(ObjectType): _meta.filter_fields = filter_fields _meta.fields = django_fields _meta.connection = connection + _meta.neomodel_filter_fields = neomodel_filter_fields super(DjangoObjectType, cls).__init_subclass_with_meta__( _meta=_meta, interfaces=interfaces, **options