graphql-python / graphene-mongo / 422

pending completion

Build # 422

Build Type

Pull #157

travis-ci

Committed by

web-flow

Commit Message

<a href="https://github.com/graphql-python/graphene-mongo/commit/<a class=hub.com/graphql-python/graphene-mongo/commit/3214c972742f5a2b2b6aae84d3c1edd094d8b221">3214c9727<a href="https://github.com/graphql-python/graphene-mongo/commit/3214c972742f5a2b2b6aae84d3c1edd094d8b221">">Merge </a><a class="double-link" href="https://github.com/graphql-python/graphene-mongo/commit/<a class="double-link" href="https://github.com/graphql-python/graphene-mongo/commit/ab2ed59b81824f7c3ffa7b747b9025c224f87f31">ab2ed59b8</a>">ab2ed59b8</a><a href="https://github.com/graphql-python/graphene-mongo/commit/3214c972742f5a2b2b6aae84d3c1edd094d8b221"> into 5d0813372">5d0813372</a>

Pull Request Pull Request #157: Feat pagination performance

Run Details

182 of 182 new or added lines in 5 files covered. (100.0%)

674 of 739 relevant lines covered (91.2%)

4.54 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.91

/graphene_mongo/fields.py

from __future__ import absolute_import

from collections import OrderedDict
from functools import partial, reduce

import graphene
import mongoengine
from bson import DBRef
from graphene import Context
from graphene.utils.str_converters import to_snake_case
from graphql import ResolveInfo
from promise import Promise
from graphql_relay import from_global_id
from graphene.relay import ConnectionField
from graphene.types.argument import to_arguments
from graphene.types.dynamic import Dynamic
from graphene.types.structures import Structure
from graphql_relay.connection.arrayconnection import cursor_to_offset
from mongoengine import QuerySet

from .advanced_types import (
    FileFieldType,
    PointFieldType,
    MultiPolygonFieldType,
    PolygonFieldType,
)
from .converter import convert_mongoengine_field, MongoEngineConversionError
from .registry import get_global_registry
from .utils import get_model_reference_fields, get_node_from_global_id, get_query_fields, find_skip_and_limit, \
    connection_from_iterables


class MongoengineConnectionField(ConnectionField):
    def __init__(self, type, *args, **kwargs):
        get_queryset = kwargs.pop("get_queryset", None)
        if get_queryset:
            assert callable(
                get_queryset
            ), "Attribute `get_queryset` on {} must be callable.".format(self)
        self._get_queryset = get_queryset
        super(MongoengineConnectionField, self).__init__(type, *args, **kwargs)

    @property
    def type(self):
        from .types import MongoengineObjectType

        _type = super(ConnectionField, self).type
        assert issubclass(
            _type, MongoengineObjectType
        ), "MongoengineConnectionField only accepts MongoengineObjectType types"
        assert _type._meta.connection, "The type {} doesn't have a connection".format(
            _type.__name__
        )
        return _type._meta.connection

    @property
    def node_type(self):
        return self.type._meta.node

    @property
    def model(self):
        return self.node_type._meta.model

    @property
    def order_by(self):
        return self.node_type._meta.order_by

    @property
    def required_fields(self):
        return tuple(set(self.node_type._meta.required_fields + self.node_type._meta.only_fields))

    @property
    def registry(self):
        return getattr(self.node_type._meta, "registry", get_global_registry())

    @property
    def args(self):
        return to_arguments(
            self._base_args or OrderedDict(),
            dict(dict(self.field_args, **self.reference_args), **self.filter_args),
        )

    @args.setter
    def args(self, args):
        self._base_args = args

    def _field_args(self, items):
        def is_filterable(k):
            """
            Remove complex columns from input args at this moment.

            Args:
                k (str): field name.
            Returns:
                bool
            """

            if not hasattr(self.model, k):
                return False
            if isinstance(getattr(self.model, k), property):
                return False
            try:
                converted = convert_mongoengine_field(
                    getattr(self.model, k), self.registry
                )
            except MongoEngineConversionError:
                return False
            if isinstance(converted, (ConnectionField, Dynamic)):
                return False
            if callable(getattr(converted, "type", None)) and isinstance(
                    converted.type(),
                    (
                            FileFieldType,
                            PointFieldType,
                            MultiPolygonFieldType,
                            graphene.Union,
                            PolygonFieldType,
                    ),
            ):
                return False
            if isinstance(converted, (graphene.List)) and issubclass(
                    getattr(converted, "_of_type", None), graphene.Union
            ):
                return False

            return True

        def get_filter_type(_type):
            """
            Returns the scalar type.
            """
            if isinstance(_type, Structure):
                return get_filter_type(_type.of_type)
            return _type()

        return {k: get_filter_type(v.type) for k, v in items if is_filterable(k)}

    @property
    def field_args(self):
        return self._field_args(self.fields.items())

    @property
    def filter_args(self):
        filter_args = dict()
        if self._type._meta.filter_fields:
            for field, filter_collection in self._type._meta.filter_fields.items():
                for each in filter_collection:
                    filter_type = getattr(
                        graphene,
                        str(self._type._meta.fields[field].type).replace("!", ""),
                    )

                    # handle special cases
                    advanced_filter_types = {
                        "in": graphene.List(filter_type),
                        "nin": graphene.List(filter_type),
                        "all": graphene.List(filter_type),
                    }

                    filter_type = advanced_filter_types.get(each, filter_type)
                    filter_args[field + "__" + each] = graphene.Argument(
                        type=filter_type
                    )

        return filter_args

    @property
    def reference_args(self):
        def get_reference_field(r, kv):
            field = kv[1]
            mongo_field = getattr(self.model, kv[0], None)
            if isinstance(
                    mongo_field,
                    (mongoengine.LazyReferenceField, mongoengine.ReferenceField),
            ):
                field = convert_mongoengine_field(mongo_field, self.registry)
            if callable(getattr(field, "get_type", None)):
                _type = field.get_type()
                if _type:
                    node = _type._type._meta
                    if "id" in node.fields and not issubclass(
                            node.model, (mongoengine.EmbeddedDocument,)
                    ):
                        r.update({kv[0]: node.fields["id"]._type.of_type()})
            return r

        return reduce(get_reference_field, self.fields.items(), {})

    @property
    def fields(self):
        return self._type._meta.fields

    def get_queryset(self, model, info, required_fields=list(), skip=None, limit=None, reversed=False, **args):
        if args:
            reference_fields = get_model_reference_fields(self.model)
            hydrated_references = {}
            for arg_name, arg in args.copy().items():
                if arg_name in reference_fields:
                    reference_obj = get_node_from_global_id(
                        reference_fields[arg_name], info, args.pop(arg_name)
                    )
                    hydrated_references[arg_name] = reference_obj
            args.update(hydrated_references)

        if self._get_queryset:
            queryset_or_filters = self._get_queryset(model, info, **args)
            if isinstance(queryset_or_filters, mongoengine.QuerySet):
                return queryset_or_filters
            else:
                args.update(queryset_or_filters)
        if limit is not None:
            if reversed:
                order_by = ""
                if self.order_by:
                    order_by = self.order_by + ",-pk"
                else:
                    order_by = "-pk"
                return model.objects(**args).no_dereference().only(*required_fields).order_by(order_by).skip(
                    skip if skip else 0).limit(limit)
            else:
                return model.objects(**args).no_dereference().only(*required_fields).order_by(self.order_by).skip(
                    skip if skip else 0).limit(limit)
        elif skip is not None:
            if reversed:
                order_by = ""
                if self.order_by:
                    order_by = self.order_by + ",-pk"
                else:
                    order_by = "-pk"
                return model.objects(**args).no_dereference().only(*required_fields).order_by(order_by).skip(
                    skip)
            else:
                return model.objects(**args).no_dereference().only(*required_fields).order_by(self.order_by).skip(
                    skip)
        return model.objects(**args).no_dereference().only(*required_fields).order_by(self.order_by)

    def default_resolver(self, _root, info, required_fields=list(), **args):
        args = args or {}

        if _root is not None:
            field_name = to_snake_case(info.field_name)
            if getattr(_root, field_name, []) is not None:
                args["pk__in"] = [r.id for r in getattr(_root, field_name, [])]

        _id = args.pop('id', None)

        if _id is not None:
            args['pk'] = from_global_id(_id)[-1]
        iterables = []
        list_length = 0
        skip = 0
        count = 0
        limit = None
        reverse = False
        if callable(getattr(self.model, "objects", None)):
            first = args.pop("first", None)
            after = cursor_to_offset(args.pop("after", None))
            last = args.pop("last", None)
            before = cursor_to_offset(args.pop("before", None))
            if after is not None:
                has_previous_page = after > 0
            elif (before is not None and last is not None):
                has_previous_page = before - last <= 0
            if "pk__in" in args and args["pk__in"]:
                count = len(args["pk__in"])
                skip, limit, reverse = find_skip_and_limit(first=first, last=last, after=after, before=before,
                                                           count=count)
                if limit:
                    if reverse:
                        args["pk__in"] = args["pk__in"][::-1][skip:skip + limit]
                    else:
                        args["pk__in"] = args["pk__in"][skip:skip + limit]
                elif skip:
                    args["pk__in"] = args["pk__in"][skip:]
                iterables = self.get_queryset(self.model, info, required_fields, **args)
                list_length = len(iterables)
                if isinstance(info, ResolveInfo):
                    if not info.context:
                        info.context = Context()
                    info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
            elif _root is None:
                count = self.get_queryset(self.model, info, required_fields, **args).count()
                if count != 0:
                    skip, limit, reverse = find_skip_and_limit(first=first, after=after, last=last, before=before,
                                                               count=count)
                    iterables = self.get_queryset(self.model, info, required_fields, skip, limit, reverse, **args)
                    list_length = len(iterables)
                    if isinstance(info, ResolveInfo):
                        if not info.context:
                            info.context = Context()
                        info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
        has_next_page = True if (0 if limit is None else limit) + (0 if skip is None else skip) < count else False
        has_previous_page = True if skip else False
        if reverse:
            iterables = list(iterables)
            iterables.reverse()
            skip = limit
        connection = connection_from_iterables(edges=iterables, start_offset=skip,
                                               has_previous_page=has_previous_page,
                                               has_next_page=has_next_page,
                                               connection_type=self.type,
                                               edge_type=self.type.Edge,
                                               pageinfo_type=graphene.PageInfo)

        connection.iterable = iterables
        connection.list_length = list_length
        return connection

    def chained_resolver(self, resolver, is_partial, root, info, **args):
        required_fields = list()
        for field in self.required_fields:
            if field in self.model._fields_ordered:
                required_fields.append(field)
        for field in get_query_fields(info):
            if to_snake_case(field) in self.model._fields_ordered:
                required_fields.append(to_snake_case(field))
        if not bool(args) or not is_partial:
            if isinstance(self.model, mongoengine.Document) or isinstance(self.model,
                                                                          mongoengine.base.metaclasses.TopLevelDocumentMetaclass):
                args_copy = args.copy()
                for arg_name, arg in args.copy().items():
                    if arg_name not in self.model._fields_ordered:
                        args_copy.pop(arg_name)
                if isinstance(info, ResolveInfo):
                    if not info.context:
                        info.context = Context()
                    info.context.queryset = self.get_queryset(self.model, info, required_fields, **args_copy)
            # XXX: Filter nested args
            resolved = resolver(root, info, **args)
            if resolved is not None:
                if isinstance(resolved, list):
                    if resolved == list():
                        return resolved
                    elif not isinstance(resolved[0], DBRef):
                        return resolved
                elif isinstance(resolved, QuerySet):
                    args.update(resolved._query)
                    return self.default_resolver(root, info, required_fields, **args)
                else:
                    return resolved
        return self.default_resolver(root, info, required_fields, **args)

    @classmethod
    def connection_resolver(cls, resolver, connection_type, root, info, **args):
        iterable = resolver(root, info, **args)
        if isinstance(connection_type, graphene.NonNull):
            connection_type = connection_type.of_type
        on_resolve = partial(cls.resolve_connection, connection_type, args)
        if Promise.is_thenable(iterable):
            return Promise.resolve(iterable).then(on_resolve)
        return on_resolve(iterable)

    def get_resolver(self, parent_resolver):
        super_resolver = self.resolver or parent_resolver
        resolver = partial(
            self.chained_resolver, super_resolver, isinstance(super_resolver, partial)
        )
        return partial(self.connection_resolver, resolver, self.type)

1	from __future__ import absolute_import	5✔
2
3	from collections import OrderedDict	5✔
4	from functools import partial, reduce	5✔
5
6	import graphene	5✔
7	import mongoengine	5✔
8	from bson import DBRef	5✔
9	from graphene import Context	5✔
10	from graphene.utils.str_converters import to_snake_case	5✔
11	from graphql import ResolveInfo	5✔
12	from promise import Promise	5✔
13	from graphql_relay import from_global_id	5✔
14	from graphene.relay import ConnectionField	5✔
15	from graphene.types.argument import to_arguments	5✔
16	from graphene.types.dynamic import Dynamic	5✔
17	from graphene.types.structures import Structure	5✔
18	from graphql_relay.connection.arrayconnection import cursor_to_offset	5✔
19	from mongoengine import QuerySet	5✔
20
21	from .advanced_types import (	5✔
22	FileFieldType,
23	PointFieldType,
24	MultiPolygonFieldType,
25	PolygonFieldType,
26	)
27	from .converter import convert_mongoengine_field, MongoEngineConversionError	5✔
28	from .registry import get_global_registry	5✔
29	from .utils import get_model_reference_fields, get_node_from_global_id, get_query_fields, find_skip_and_limit, \	5✔
30	connection_from_iterables
31
32
33	class MongoengineConnectionField(ConnectionField):	5✔
34	def __init__(self, type, args, *kwargs):	5✔
35	get_queryset = kwargs.pop("get_queryset", None)	5✔
36	if get_queryset:	5✔
37	assert callable(	5✔
38	get_queryset
39	), "Attribute `get_queryset` on {} must be callable.".format(self)
40	self._get_queryset = get_queryset	5✔
41	super(MongoengineConnectionField, self).__init__(type, args, *kwargs)	5✔
42
43	@property	5✔
44	def type(self):	1✔
45	from .types import MongoengineObjectType	5✔
46
47	_type = super(ConnectionField, self).type	5✔
48	assert issubclass(	5✔
49	_type, MongoengineObjectType
50	), "MongoengineConnectionField only accepts MongoengineObjectType types"
51	assert _type._meta.connection, "The type {} doesn't have a connection".format(	5✔
52	_type.__name__
53	)
54	return _type._meta.connection	5✔
55
56	@property	5✔
57	def node_type(self):	1✔
58	return self.type._meta.node	5✔
59
60	@property	5✔
61	def model(self):	1✔
62	return self.node_type._meta.model	5✔
63
64	@property	5✔
65	def order_by(self):	1✔
66	return self.node_type._meta.order_by	5✔
67
68	@property	5✔
69	def required_fields(self):	1✔
70	return tuple(set(self.node_type._meta.required_fields + self.node_type._meta.only_fields))	5✔
71
72	@property	5✔
73	def registry(self):	1✔
74	return getattr(self.node_type._meta, "registry", get_global_registry())	5✔
75
76	@property	5✔
77	def args(self):	1✔
78	return to_arguments(	5✔
79	self._base_args or OrderedDict(),
80	dict(dict(self.field_args, self.reference_args), self.filter_args),
81	)
82
83	@args.setter	5✔
84	def args(self, args):	1✔
85	self._base_args = args	5✔
86
87	def _field_args(self, items):	5✔
88	def is_filterable(k):	5✔
89	"""
90	Remove complex columns from input args at this moment.
91
92	Args:
93	k (str): field name.
94	Returns:
95	bool
96	"""
97
98	if not hasattr(self.model, k):	5✔
99	return False	5✔
100	if isinstance(getattr(self.model, k), property):	5✔
101	return False	5✔
102	try:	5✔
103	converted = convert_mongoengine_field(	5✔
104	getattr(self.model, k), self.registry
105	)
106	except MongoEngineConversionError:	5✔
107	return False	5✔
108	if isinstance(converted, (ConnectionField, Dynamic)):	5✔
109	return False	5✔
110	if callable(getattr(converted, "type", None)) and isinstance(	5✔
111	converted.type(),
112	(
113	FileFieldType,
114	PointFieldType,
115	MultiPolygonFieldType,
116	graphene.Union,
117	PolygonFieldType,
118	),
119	):
120	return False	5✔
121	if isinstance(converted, (graphene.List)) and issubclass(	5✔
122	getattr(converted, "_of_type", None), graphene.Union
123	):
124	return False	5✔
125
126	return True	5✔
127
128	def get_filter_type(_type):	5✔
129	"""
130	Returns the scalar type.
131	"""
132	if isinstance(_type, Structure):	5✔
133	return get_filter_type(_type.of_type)	5✔
134	return _type()	5✔
135
136	return {k: get_filter_type(v.type) for k, v in items if is_filterable(k)}	5✔
137
138	@property	5✔
139	def field_args(self):	1✔
140	return self._field_args(self.fields.items())	5✔
141
142	@property	5✔
143	def filter_args(self):	1✔
144	filter_args = dict()	5✔
145	if self._type._meta.filter_fields:	5✔
146	for field, filter_collection in self._type._meta.filter_fields.items():	5✔
147	for each in filter_collection:	5✔
148	filter_type = getattr(	5✔
149	graphene,
150	str(self._type._meta.fields[field].type).replace("!", ""),
151	)
152
153	# handle special cases
154	advanced_filter_types = {	5✔
155	"in": graphene.List(filter_type),
156	"nin": graphene.List(filter_type),
157	"all": graphene.List(filter_type),
158	}
159
160	filter_type = advanced_filter_types.get(each, filter_type)	5✔
161	filter_args[field + "__" + each] = graphene.Argument(	5✔
162	type=filter_type
163	)
164
165	return filter_args	5✔
166
167	@property	5✔
168	def reference_args(self):	1✔
169	def get_reference_field(r, kv):	5✔
170	field = kv[1]	5✔
171	mongo_field = getattr(self.model, kv[0], None)	5✔
172	if isinstance(	5✔
173	mongo_field,
174	(mongoengine.LazyReferenceField, mongoengine.ReferenceField),
175	):
176	field = convert_mongoengine_field(mongo_field, self.registry)	5✔
177	if callable(getattr(field, "get_type", None)):	5✔
178	_type = field.get_type()	5✔
179	if _type:	5✔
180	node = _type._type._meta	5✔
181	if "id" in node.fields and not issubclass(	5✔
182	node.model, (mongoengine.EmbeddedDocument,)
183	):
184	r.update({kv[0]: node.fields["id"]._type.of_type()})	5✔
185	return r	5✔
186
187	return reduce(get_reference_field, self.fields.items(), {})	5✔
188
189	@property	5✔
190	def fields(self):	1✔
191	return self._type._meta.fields	5✔
192
193	def get_queryset(self, model, info, required_fields=list(), skip=None, limit=None, reversed=False, **args):	5✔
194	if args:	5✔
195	reference_fields = get_model_reference_fields(self.model)	5✔
196	hydrated_references = {}	5✔
197	for arg_name, arg in args.copy().items():	5✔
198	if arg_name in reference_fields:	5✔
199	reference_obj = get_node_from_global_id(	5✔
200	reference_fields[arg_name], info, args.pop(arg_name)
201	)
202	hydrated_references[arg_name] = reference_obj	5✔
203	args.update(hydrated_references)	5✔
204
205	if self._get_queryset:	5✔
206	queryset_or_filters = self._get_queryset(model, info, **args)	5✔
207	if isinstance(queryset_or_filters, mongoengine.QuerySet):	5✔
208	return queryset_or_filters	5✔
209	else:
210	args.update(queryset_or_filters)	5✔
211	if limit is not None:	5✔
212	if reversed:	5✔
213	order_by = ""	5✔
214	if self.order_by:	5✔
215	order_by = self.order_by + ",-pk"	×
216	else:
217	order_by = "-pk"	5✔
218	return model.objects(*args).no_dereference().only(required_fields).order_by(order_by).skip(	5✔
219	skip if skip else 0).limit(limit)
220	else:
221	return model.objects(*args).no_dereference().only(required_fields).order_by(self.order_by).skip(	5✔
222	skip if skip else 0).limit(limit)
223	elif skip is not None:	5✔
224	if reversed:	5✔
225	order_by = ""	×
226	if self.order_by:	×
227	order_by = self.order_by + ",-pk"	×
228	else:
229	order_by = "-pk"	×
230	return model.objects(*args).no_dereference().only(required_fields).order_by(order_by).skip(	×
231	skip)
232	else:
233	return model.objects(*args).no_dereference().only(required_fields).order_by(self.order_by).skip(	5✔
234	skip)
235	return model.objects(*args).no_dereference().only(required_fields).order_by(self.order_by)	5✔
236
237	def default_resolver(self, _root, info, required_fields=list(), **args):	5✔
238	args = args or {}	5✔
239
240	if _root is not None:	5✔
241	field_name = to_snake_case(info.field_name)	5✔
242	if getattr(_root, field_name, []) is not None:	5✔
243	args["pk__in"] = [r.id for r in getattr(_root, field_name, [])]	5✔
244
245	_id = args.pop('id', None)	5✔
246
247	if _id is not None:	5✔
248	args['pk'] = from_global_id(_id)[-1]	5✔
249	iterables = []	5✔
250	list_length = 0	5✔
251	skip = 0	5✔
252	count = 0	5✔
253	limit = None	5✔
254	reverse = False	5✔
255	if callable(getattr(self.model, "objects", None)):	5✔
256	first = args.pop("first", None)	5✔
257	after = cursor_to_offset(args.pop("after", None))	5✔
258	last = args.pop("last", None)	5✔
259	before = cursor_to_offset(args.pop("before", None))	5✔
260	if after is not None:	5✔
261	has_previous_page = after > 0	5✔
262	elif (before is not None and last is not None):	5✔
263	has_previous_page = before - last <= 0	×
264	if "pk__in" in args and args["pk__in"]:	5✔
265	count = len(args["pk__in"])	5✔
266	skip, limit, reverse = find_skip_and_limit(first=first, last=last, after=after, before=before,	5✔
267	count=count)
268	if limit:	5✔
269	if reverse:	5✔
270	args["pk__in"] = args["pk__in"][::-1][skip:skip + limit]	×
271	else:
272	args["pk__in"] = args["pk__in"][skip:skip + limit]	5✔
273	elif skip:	5✔
274	args["pk__in"] = args["pk__in"][skip:]	×
275	iterables = self.get_queryset(self.model, info, required_fields, **args)	5✔
276	list_length = len(iterables)	5✔
277	if isinstance(info, ResolveInfo):	5✔
278	if not info.context:	5✔
279	info.context = Context()	5✔
280	info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)	5✔
281	elif _root is None:	5✔
282	count = self.get_queryset(self.model, info, required_fields, **args).count()	5✔
283	if count != 0:	5✔
284	skip, limit, reverse = find_skip_and_limit(first=first, after=after, last=last, before=before,	5✔
285	count=count)
286	iterables = self.get_queryset(self.model, info, required_fields, skip, limit, reverse, **args)	5✔
287	list_length = len(iterables)	5✔
288	if isinstance(info, ResolveInfo):	5✔
289	if not info.context:	5✔
290	info.context = Context()	5✔
291	info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)	5✔
292	has_next_page = True if (0 if limit is None else limit) + (0 if skip is None else skip) < count else False	5✔
293	has_previous_page = True if skip else False	5✔
294	if reverse:	5✔
295	iterables = list(iterables)	5✔
296	iterables.reverse()	5✔
297	skip = limit	5✔
298	connection = connection_from_iterables(edges=iterables, start_offset=skip,	5✔
299	has_previous_page=has_previous_page,
300	has_next_page=has_next_page,
301	connection_type=self.type,
302	edge_type=self.type.Edge,
303	pageinfo_type=graphene.PageInfo)
304
305	connection.iterable = iterables	5✔
306	connection.list_length = list_length	5✔
307	return connection	5✔
308
309	def chained_resolver(self, resolver, is_partial, root, info, **args):	5✔
310	required_fields = list()	5✔
311	for field in self.required_fields:	5✔
312	if field in self.model._fields_ordered:	×
313	required_fields.append(field)	×
314	for field in get_query_fields(info):	5✔
315	if to_snake_case(field) in self.model._fields_ordered:	5✔
316	required_fields.append(to_snake_case(field))	5✔
317	if not bool(args) or not is_partial:	5✔
318	if isinstance(self.model, mongoengine.Document) or isinstance(self.model,	5✔
319	mongoengine.base.metaclasses.TopLevelDocumentMetaclass):
320	args_copy = args.copy()	5✔
321	for arg_name, arg in args.copy().items():	5✔
322	if arg_name not in self.model._fields_ordered:	5✔
323	args_copy.pop(arg_name)	5✔
324	if isinstance(info, ResolveInfo):	5✔
325	if not info.context:	5✔
326	info.context = Context()	5✔
327	info.context.queryset = self.get_queryset(self.model, info, required_fields, **args_copy)	5✔
328	# XXX: Filter nested args
329	resolved = resolver(root, info, **args)	5✔
330	if resolved is not None:	5✔
331	if isinstance(resolved, list):	5✔
332	if resolved == list():	5✔
333	return resolved	5✔
334	elif not isinstance(resolved[0], DBRef):	5✔
335	return resolved	5✔
336	elif isinstance(resolved, QuerySet):	5✔
337	args.update(resolved._query)	×
338	return self.default_resolver(root, info, required_fields, **args)	×
339	else:
340	return resolved	5✔
341	return self.default_resolver(root, info, required_fields, **args)	5✔
342
343	@classmethod	5✔
344	def connection_resolver(cls, resolver, connection_type, root, info, **args):	1✔
345	iterable = resolver(root, info, **args)	5✔
346	if isinstance(connection_type, graphene.NonNull):	5✔
347	connection_type = connection_type.of_type	×
348	on_resolve = partial(cls.resolve_connection, connection_type, args)	5✔
349	if Promise.is_thenable(iterable):	5✔
350	return Promise.resolve(iterable).then(on_resolve)	5✔
351	return on_resolve(iterable)	5✔
352
353	def get_resolver(self, parent_resolver):	5✔
354	super_resolver = self.resolver or parent_resolver	5✔
355	resolver = partial(	5✔
356	self.chained_resolver, super_resolver, isinstance(super_resolver, partial)
357	)
358	return partial(self.connection_resolver, resolver, self.type)	5✔

graphql-python / graphene-mongo / 422

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous