mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-25 11:04:11 +03:00
Allow string references in DjangoListField (#885)
* Allow passing string references to DjangoListField * Refactor logic to work with string imports
This commit is contained in:
parent
348fcf37a0
commit
c8a56f8857
|
@ -1,5 +1,6 @@
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
|
import six
|
||||||
from django.db.models.query import QuerySet
|
from django.db.models.query import QuerySet
|
||||||
from graphql_relay.connection.arrayconnection import connection_from_list_slice
|
from graphql_relay.connection.arrayconnection import connection_from_list_slice
|
||||||
from promise import Promise
|
from promise import Promise
|
||||||
|
@ -19,19 +20,23 @@ class DjangoListField(Field):
|
||||||
if isinstance(_type, NonNull):
|
if isinstance(_type, NonNull):
|
||||||
_type = _type.of_type
|
_type = _type.of_type
|
||||||
|
|
||||||
assert issubclass(
|
|
||||||
_type, DjangoObjectType
|
|
||||||
), "DjangoListField only accepts DjangoObjectType types"
|
|
||||||
|
|
||||||
# Django would never return a Set of None vvvvvvv
|
# Django would never return a Set of None vvvvvvv
|
||||||
super(DjangoListField, self).__init__(List(NonNull(_type)), *args, **kwargs)
|
super(DjangoListField, self).__init__(List(NonNull(_type)), *args, **kwargs)
|
||||||
|
|
||||||
|
assert issubclass(
|
||||||
|
self._underlying_type, DjangoObjectType
|
||||||
|
), "DjangoListField only accepts DjangoObjectType types"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _underlying_type(self):
|
||||||
|
_type = self._type
|
||||||
|
while hasattr(_type, "of_type"):
|
||||||
|
_type = _type.of_type
|
||||||
|
return _type
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def model(self):
|
def model(self):
|
||||||
_type = self.type.of_type
|
return self._underlying_type._meta.model
|
||||||
if isinstance(_type, NonNull):
|
|
||||||
_type = _type.of_type
|
|
||||||
return _type._meta.model
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def list_resolver(django_object_type, resolver, root, info, **args):
|
def list_resolver(django_object_type, resolver, root, info, **args):
|
||||||
|
|
|
@ -19,6 +19,12 @@ class TestDjangoListField:
|
||||||
with pytest.raises(AssertionError):
|
with pytest.raises(AssertionError):
|
||||||
list_field = DjangoListField(TestType)
|
list_field = DjangoListField(TestType)
|
||||||
|
|
||||||
|
def test_only_import_paths(self):
|
||||||
|
list_field = DjangoListField("graphene_django.tests.schema.Human")
|
||||||
|
from .schema import Human
|
||||||
|
|
||||||
|
assert list_field._type.of_type.of_type is Human
|
||||||
|
|
||||||
def test_non_null_type(self):
|
def test_non_null_type(self):
|
||||||
class Reporter(DjangoObjectType):
|
class Reporter(DjangoObjectType):
|
||||||
class Meta:
|
class Meta:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user