Merge branch 'feat-retrieving-queried-fields-only' into feat-pagination-performance

Author: arunsureshkumar

graphene_mongo/converter.py

@@ -221,7 +221,12 @@ def reference_resolver(root, *args, **kwargs):
         return None
 
     if isinstance(field, mongoengine.GenericReferenceField):
-        return graphene.Field(_union, resolver=reference_resolver,
+        field_resolver = None
+        if field.db_field is not None:
+            resolver_function = getattr(_union, "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,
                               description=get_field_description(field, registry))
 
     return graphene.Field(_union)
@@ -259,14 +264,20 @@ def dynamic_type():
         _type = registry.get_type_for_model(model)
         if not _type:
             return None
-        elif isinstance(field, mongoengine.ReferenceField):
-            return graphene.Field(_type, resolver=reference_resolver,
+        if isinstance(field, mongoengine.EmbeddedDocumentField):
+            return graphene.Field(_type,
                                   description=get_field_description(field, registry))
-        elif isinstance(field, mongoengine.CachedReferenceField):
-            return graphene.Field(_type, resolver=cached_reference_resolver,
+        field_resolver = None
+        if field.db_field is not None:
+            resolver_function = getattr(_type, "resolve_" + field.db_field, None)
+            if resolver_function and callable(resolver_function):
+                field_resolver = resolver_function
+        if isinstance(field, mongoengine.ReferenceField):
+            return graphene.Field(_type, resolver=field_resolver if field_resolver else reference_resolver,
+                                  description=get_field_description(field, registry))
+        else:
+            return graphene.Field(_type, resolver=field_resolver if field_resolver else cached_reference_resolver(),
                                   description=get_field_description(field, registry))
-        return graphene.Field(_type,
-                              description=get_field_description(field, registry))
 
     return graphene.Dynamic(dynamic_type)
 
@@ -288,12 +299,16 @@ def lazy_resolver(root, *args, **kwargs):
 
     def dynamic_type():
         _type = registry.get_type_for_model(model)
-
         if not _type:
             return None
+        field_resolver = None
+        if field.db_field is not None:
+            resolver_function = getattr(_type, "resolve_" + field.db_field, None)
+            if resolver_function and callable(resolver_function):
+                field_resolver = resolver_function
         return graphene.Field(
             _type,
-            resolver=lazy_resolver,
+            resolver=field_resolver if field_resolver else lazy_resolver,
             description=get_field_description(field, registry),
         )