mirror of
https://github.com/graphql-python/graphene-django.git
synced 2025-07-12 17:22:20 +03:00
Refactor logic to work with string imports
This commit is contained in:
parent
38a0d4c4f2
commit
a516c5dc0c
|
@ -20,20 +20,23 @@ class DjangoListField(Field):
|
|||
if isinstance(_type, NonNull):
|
||||
_type = _type.of_type
|
||||
|
||||
if not isinstance(_type, six.string_types):
|
||||
assert issubclass(
|
||||
_type, DjangoObjectType
|
||||
), "DjangoListField only accepts DjangoObjectType types"
|
||||
|
||||
# Django would never return a Set of None vvvvvvv
|
||||
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
|
||||
def model(self):
|
||||
_type = self.type.of_type
|
||||
if isinstance(_type, NonNull):
|
||||
_type = _type.of_type
|
||||
return _type._meta.model
|
||||
return self._underlying_type._meta.model
|
||||
|
||||
@staticmethod
|
||||
def list_resolver(django_object_type, resolver, root, info, **args):
|
||||
|
|
|
@ -20,8 +20,10 @@ class TestDjangoListField:
|
|||
list_field = DjangoListField(TestType)
|
||||
|
||||
def test_only_import_paths(self):
|
||||
list_field = DjangoListField("graphene_django.tests.models.Reporter")
|
||||
assert list_field._type.of_type.of_type is ReporterModel
|
||||
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):
|
||||
class Reporter(DjangoObjectType):
|
||||
|
|
Loading…
Reference in New Issue
Block a user