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

Optimize DjangoConnectionField too

Browse Source
This commit is contained in:
Jacob Foster 2017-07-19 14:20:52 -05:00
parent f585518cc1
commit b206278e95
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphene_django/fields.py
View File

@ -8,6 +8,7 @@ from graphene.types import Field, List
from graphene.relay import ConnectionField, PageInfo from graphene.relay import ConnectionField, PageInfo
from graphql_relay.connection.arrayconnection import connection_from_list_slice from graphql_relay.connection.arrayconnection import connection_from_list_slice
from .optimization import optimize_queryset
from .settings import graphene_settings from .settings import graphene_settings
from .utils import DJANGO_FILTER_INSTALLED, maybe_queryset from .utils import DJANGO_FILTER_INSTALLED, maybe_queryset
@ -62,7 +63,7 @@ class DjangoConnectionField(ConnectionField):
return queryset & default_queryset return queryset & default_queryset
@classmethod @classmethod
def resolve_connection(cls, connection, default_manager, args, iterable): def resolve_connection(cls, connection, default_manager, args, info, iterable):
if iterable is None: if iterable is None:
iterable = default_manager iterable = default_manager
iterable = maybe_queryset(iterable) iterable = maybe_queryset(iterable)
@ -70,6 +71,7 @@ class DjangoConnectionField(ConnectionField):
if iterable is not default_manager: if iterable is not default_manager:
default_queryset = maybe_queryset(default_manager) default_queryset = maybe_queryset(default_manager)
iterable = cls.merge_querysets(default_queryset, iterable) iterable = cls.merge_querysets(default_queryset, iterable)
iterable = optimize_queryset(default_manager.model, iterable, info.field_asts[0])
_len = iterable.count() _len = iterable.count()
else: else:
_len = len(iterable) _len = len(iterable)
@ -112,7 +114,7 @@ class DjangoConnectionField(ConnectionField):
args['last'] = min(last, max_limit) args['last'] = min(last, max_limit)
iterable = resolver(root, args, context, info) iterable = resolver(root, args, context, info)
on_resolve = partial(cls.resolve_connection, connection, default_manager, args) on_resolve = partial(cls.resolve_connection, connection, default_manager, args, info)
if Promise.is_thenable(iterable): if Promise.is_thenable(iterable):
return Promise.resolve(iterable).then(on_resolve) return Promise.resolve(iterable).then(on_resolve)