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 django.db.models.query import QuerySet
from promise import Promise from promise import Promise
from neomodel import NodeSet
from graphene.types import Field, List from graphene.types import Field, List
from graphene.relay import ConnectionField, PageInfo from graphene.relay import ConnectionField, PageInfo
@ -80,7 +81,7 @@ class DjangoConnectionField(ConnectionField):
if iterable is None: if iterable is None:
iterable = default_manager iterable = default_manager
iterable = maybe_queryset(iterable) iterable = maybe_queryset(iterable)
if isinstance(iterable, QuerySet): if isinstance(iterable, NodeSet):
if iterable is not default_manager: if iterable is not default_manager:
default_queryset = maybe_queryset(default_manager) default_queryset = maybe_queryset(default_manager)
iterable = cls.merge_querysets(default_queryset, iterable) iterable = cls.merge_querysets(default_queryset, iterable)

View File

@ -54,6 +54,7 @@ class DjangoObjectType(ObjectType):
only_fields=(), only_fields=(),
exclude_fields=(), exclude_fields=(),
filter_fields=None, filter_fields=None,
neomodel_filter_fields=None,
connection=None, connection=None,
connection_class=None, connection_class=None,
use_connection=None, use_connection=None,
@ -106,6 +107,7 @@ class DjangoObjectType(ObjectType):
_meta.filter_fields = filter_fields _meta.filter_fields = filter_fields
_meta.fields = django_fields _meta.fields = django_fields
_meta.connection = connection _meta.connection = connection
_meta.neomodel_filter_fields = neomodel_filter_fields
super(DjangoObjectType, cls).__init_subclass_with_meta__( super(DjangoObjectType, cls).__init_subclass_with_meta__(
_meta=_meta, interfaces=interfaces, **options _meta=_meta, interfaces=interfaces, **options