support filter agrs in default reference resolvers

Author: arunsureshkumar

graphene_mongo/converter.py

@@ -112,9 +112,14 @@ def get_reference_objects(*args, **kwargs):
                 document_field = convert_mongoengine_field(document_field, registry)
                 document_field_type = document_field.get_type().type
                 queried_fields = list()
+                filter_args = list()
+                if document_field_type._meta.filter_fields:
+                    for key, values in document_field_type._meta.filter_fields.items():
+                        for each in values:
+                            filter_args.append(key + "__" + each)
                 for each in get_query_fields(args[0][3][0])[document_field_type._meta.name].keys():
                     item = to_snake_case(each)
-                    if item in document._fields_ordered:
+                    if item in document._fields_ordered + tuple(filter_args):
                         queried_fields.append(item)
                 return document.objects().no_dereference().only(
                     *set(list(document_field_type._meta.required_fields) + queried_fields)).filter(pk__in=args[0][1])
@@ -229,12 +234,17 @@ def reference_resolver(root, *args, **kwargs):
             document_field = mongoengine.ReferenceField(document)
             document_field = convert_mongoengine_field(document_field, registry)
             _type = document_field.get_type().type
+            filter_args = list()
+            if _type._meta.filter_fields:
+                for key, values in _type._meta.filter_fields.items():
+                    for each in values:
+                        filter_args.append(key + "__" + each)
             querying_types = list(get_query_fields(args[0]).keys())
             if _type.__name__ in querying_types:
                 queried_fields = list()
                 for each in get_query_fields(args[0]).keys():
                     item = to_snake_case(each)
-                    if item in document._fields_ordered:
+                    if item in document._fields_ordered + tuple(filter_args):
                         queried_fields.append(item)
                 return document.objects().no_dereference().only(*list(
                     set(list(_type._meta.required_fields) + queried_fields))).get(
@@ -247,7 +257,8 @@ def reference_resolver(root, *args, **kwargs):
         required = False
         if field.db_field is not None:
             required = field.required
-            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
+            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field,
+                                        None)
             if resolver_function and callable(resolver_function):
                 field_resolver = resolver_function
         return graphene.Field(_union, resolver=field_resolver if field_resolver else reference_resolver,
@@ -266,11 +277,16 @@ def reference_resolver(root, *args, **kwargs):
         document = getattr(root, field.name or field.db_name)
         if document:
             queried_fields = list()
+            _type = registry.get_type_for_model(field.document_type)
+            filter_args = list()
+            if _type._meta.filter_fields:
+                for key, values in _type._meta.filter_fields.items():
+                    for each in values:
+                        filter_args.append(key + "__" + each)
             for each in get_query_fields(args[0]).keys():
                 item = to_snake_case(each)
-                if item in field.document_type._fields_ordered:
+                if item in field.document_type._fields_ordered + tuple(filter_args):
                     queried_fields.append(item)
-            _type = registry.get_type_for_model(field.document_type)
             return field.document_type.objects().no_dereference().only(
                 *(set(list(_type._meta.required_fields) + queried_fields))).get(
                 pk=document.id)
@@ -279,11 +295,16 @@ def reference_resolver(root, *args, **kwargs):
     def cached_reference_resolver(root, *args, **kwargs):
         if field:
             queried_fields = list()
+            _type = registry.get_type_for_model(field.document_type)
+            filter_args = list()
+            if _type._meta.filter_fields:
+                for key, values in _type._meta.filter_fields.items():
+                    for each in values:
+                        filter_args.append(key + "__" + each)
             for each in get_query_fields(args[0]).keys():
                 item = to_snake_case(each)
-                if item in field.document_type._fields_ordered:
+                if item in field.document_type._fields_ordered + tuple(filter_args):
                     queried_fields.append(item)
-            _type = registry.get_type_for_model(field.document_type)
             return field.document_type.objects().no_dereference().only(
                 *(set(
                     list(_type._meta.required_fields) + queried_fields))).get(
@@ -301,7 +322,8 @@ def dynamic_type():
         required = False
         if field.db_field is not None:
             required = field.required
-            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
+            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field,
+                                        None)
             if resolver_function and callable(resolver_function):
                 field_resolver = resolver_function
         if isinstance(field, mongoengine.ReferenceField):
@@ -322,11 +344,16 @@ def lazy_resolver(root, *args, **kwargs):
         document = getattr(root, field.name or field.db_name)
         if document:
             queried_fields = list()
+            _type = registry.get_type_for_model(document.document_type)
+            filter_args = list()
+            if _type._meta.filter_fields:
+                for key, values in _type._meta.filter_fields.items():
+                    for each in values:
+                        filter_args.append(key + "__" + each)
             for each in get_query_fields(args[0]).keys():
                 item = to_snake_case(each)
-                if item in document.document_type._fields_ordered:
+                if item in document.document_type._fields_ordered + tuple(filter_args):
                     queried_fields.append(item)
-            _type = registry.get_type_for_model(document.document_type)
             return document.document_type.objects().no_dereference().only(
                 *(set((list(_type._meta.required_fields) + queried_fields)))).get(
                 pk=document.pk)
@@ -340,7 +367,8 @@ def dynamic_type():
         required = False
         if field.db_field is not None:
             required = field.required
-            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
+            resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field,
+                                        None)
             if resolver_function and callable(resolver_function):
                 field_resolver = resolver_function
         return graphene.Field(