fix(ConnectionField): Priorities pk__in filter for finding count of nested connection field

Author: abhinand-c

graphene_mongo/fields.py

@@ -424,7 +424,24 @@ def default_resolver(self, _root, info, required_fields=None, resolved=None, **a
             list_length = len(iterables)
 
         elif callable(getattr(self.model, "objects", None)):
-            if (
+            if "pk__in" in args and args["pk__in"]:
+                count = len(args["pk__in"])
+                skip, limit = find_skip_and_limit(
+                    first=first, last=last, after=after, before=before, count=count
+                )
+                if limit:
+                    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, GraphQLResolveInfo):
+                    if not info.context:
+                        info = info._replace(context=Context())
+                    info.context.queryset = self.get_queryset(
+                        self.model, info, required_fields, **args
+                    )
+            elif (
                 _root is None
                 or args
                 or isinstance(getattr(_root, field_name, []), MongoengineConnectionField)
@@ -486,24 +503,6 @@ def default_resolver(self, _root, info, required_fields=None, resolved=None, **a
                             self.model, info, required_fields, **args
                         )
 
-            elif "pk__in" in args and args["pk__in"]:
-                count = len(args["pk__in"])
-                skip, limit = find_skip_and_limit(
-                    first=first, last=last, after=after, before=before, count=count
-                )
-                if limit:
-                    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, GraphQLResolveInfo):
-                    if not info.context:
-                        info = info._replace(context=Context())
-                    info.context.queryset = self.get_queryset(
-                        self.model, info, required_fields, **args
-                    )
-
         elif _root is not None:
             field_name = to_snake_case(info.field_name)
             items = getattr(_root, field_name, [])

graphene_mongo/fields_async.py

@@ -150,7 +150,25 @@ async def default_resolver(self, _root, info, required_fields=None, resolved=Non
             list_length = len(iterables)
 
         elif callable(getattr(self.model, "objects", None)):
-            if (
+            if "pk__in" in args and args["pk__in"]:
+                count = len(args["pk__in"])
+                skip, limit = find_skip_and_limit(
+                    first=first, last=last, after=after, before=before, count=count
+                )
+                if limit:
+                    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)
+                iterables = await sync_to_async(list)(iterables)
+                list_length = len(iterables)
+                if isinstance(info, GraphQLResolveInfo):
+                    if not info.context:
+                        info = info._replace(context=Context())
+                    info.context.queryset = self.get_queryset(
+                        self.model, info, required_fields, **args
+                    )
+            elif (
                 _root is None
                 or args
                 or isinstance(getattr(_root, field_name, []), AsyncMongoengineConnectionField)
@@ -206,25 +224,6 @@ async def default_resolver(self, _root, info, required_fields=None, resolved=Non
                             self.model, info, required_fields, **args
                         )
 
-            elif "pk__in" in args and args["pk__in"]:
-                count = len(args["pk__in"])
-                skip, limit = find_skip_and_limit(
-                    first=first, last=last, after=after, before=before, count=count
-                )
-                if limit:
-                    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)
-                iterables = await sync_to_async(list)(iterables)
-                list_length = len(iterables)
-                if isinstance(info, GraphQLResolveInfo):
-                    if not info.context:
-                        info = info._replace(context=Context())
-                    info.context.queryset = self.get_queryset(
-                        self.model, info, required_fields, **args
-                    )
-
         elif _root is not None:
             field_name = to_snake_case(info.field_name)
             items = getattr(_root, field_name, [])