added neomodel filter fields

This commit is contained in:
Mardanov Timur Rustemovich 2018-11-26 23:20:10 +03:00
parent 05093d862f
commit f7e5d8bb39
2 changed files with 4 additions and 1 deletions

View File

@ -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)

View File

@ -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