graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-11 16:52:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use max_limit first, if set.

Browse Source
This commit is contained in:
Paul Craciunoiu 2020-05-19 13:17:27 -06:00
parent d804fe48f2
commit 47a3ac2758
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
graphene_django/fields.py
View File

@ -127,12 +127,12 @@ 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, iterable, max_limit=None):
iterable = maybe_queryset(iterable) iterable = maybe_queryset(iterable)
if isinstance(iterable, QuerySet): if isinstance(iterable, QuerySet):
_len = iterable.count() _len = max_limit or iterable.count()
else: else:
_len = len(iterable) _len = max_limit or len(iterable)
connection = connection_from_list_slice( connection = connection_from_list_slice(
iterable, iterable,
args, args,
@ -189,7 +189,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, max_limit=max_limit)
if Promise.is_thenable(iterable): if Promise.is_thenable(iterable):
return Promise.resolve(iterable).then(on_resolve) return Promise.resolve(iterable).then(on_resolve)