From be2e7111b789f1aca8fe3e660330dcce76991dad Mon Sep 17 00:00:00 2001 From: Mardanov Timur Rustemovich Date: Mon, 19 Nov 2018 21:03:15 +0300 Subject: [PATCH] install --- graphene_django/utils.py | 5 ++--- setup.py | 3 +++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/graphene_django/utils.py b/graphene_django/utils.py index 560f604..9e0f9ec 100644 --- a/graphene_django/utils.py +++ b/graphene_django/utils.py @@ -1,8 +1,7 @@ import inspect from django.db import models -from django.db.models.manager import Manager - +from neomodel import NodeSet # from graphene.utils import LazyList @@ -34,7 +33,7 @@ def get_reverse_fields(model, local_field_names): def maybe_queryset(value): - if isinstance(value, Manager): + if isinstance(value, NodeSet): value = value.get_queryset() return value diff --git a/setup.py b/setup.py index 3431cd5..e025a6e 100644 --- a/setup.py +++ b/setup.py @@ -12,6 +12,7 @@ with open("graphene_django/__init__.py", "rb") as f: rest_framework_require = ["djangorestframework>=3.6.3"] +neomodel_require = ["neomodel>=3.3.0", ] tests_require = [ "pytest>=3.6.3", @@ -54,10 +55,12 @@ setup( "Django>=1.11", "singledispatch>=3.4.0.3", "promise>=2.1", + *neomodel_require, ], setup_requires=["pytest-runner"], tests_require=tests_require, rest_framework_require=rest_framework_require, + neomodel_require=neomodel_require, extras_require={"test": tests_require, "rest_framework": rest_framework_require}, include_package_data=True, zip_safe=False,