Fixing fields

This commit is contained in:
ariel1899 2020-04-20 12:27:30 -04:00
parent ef54b43ccc
commit 9fd220fa60

View File

@ -118,7 +118,11 @@ class DjangoConnectionField(ConnectionField):
return connection._meta.node.get_queryset(queryset, info) return connection._meta.node.get_queryset(queryset, info)
@classmethod @classmethod
def resolve_connection(cls, connection, args, iterable): def resolve_connection(cls, connection, args, default_manager, iterable):
if iterable is None:
iterable = default_manager
iterable = maybe_queryset(iterable) iterable = maybe_queryset(iterable)
if isinstance(iterable, QuerySet): if isinstance(iterable, QuerySet):
_len = iterable.count() _len = iterable.count()
@ -180,7 +184,7 @@ class DjangoConnectionField(ConnectionField):
# thus the iterable gets refiltered by resolve_queryset # thus the iterable gets refiltered by resolve_queryset
# but iterable might be promise # but iterable might be promise
iterable = queryset_resolver(connection, iterable, info, args) iterable = queryset_resolver(connection, iterable, info, args)
on_resolve = partial(cls.resolve_connection, connection, args) on_resolve = partial(cls.resolve_connection, connection, args, default_manager)
if Promise.is_thenable(iterable): if Promise.is_thenable(iterable):
return Promise.resolve(iterable).then(on_resolve) return Promise.resolve(iterable).then(on_resolve)