Merge pull request #164 from arunsureshkumar/support-generic-reference-field-in-args

Support generic reference field in args

Author: abawchen

graphene_mongo/fields.py

@@ -10,6 +10,7 @@
 from graphene.types.utils import get_type
 from graphene.utils.str_converters import to_snake_case
 from graphql import ResolveInfo
+from mongoengine.base import get_document
 from promise import Promise
 from graphql_relay import from_global_id
 from graphene.relay import ConnectionField
@@ -177,6 +178,9 @@ def get_reference_field(r, kv):
                     (mongoengine.LazyReferenceField, mongoengine.ReferenceField),
             ):
                 field = convert_mongoengine_field(mongo_field, self.registry)
+            if isinstance(mongo_field, mongoengine.GenericReferenceField):
+                r.update({kv[0]: graphene.ID()})
+                return r
             if callable(getattr(field, "get_type", None)):
                 _type = field.get_type()
                 if _type:
@@ -207,6 +211,15 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                     except TypeError:
                         reference_obj = reference_fields[arg_name].document_type(pk=arg)
                     hydrated_references[arg_name] = reference_obj
+                elif arg_name in self.model._fields_ordered and isinstance(getattr(self.model, arg_name),
+                                                                           mongoengine.fields.GenericReferenceField):
+                    try:
+                        reference_obj = get_document(self.registry._registry_string_map[from_global_id(arg)[0]])(
+                            pk=from_global_id(arg)[1])
+                    except TypeError:
+                        reference_obj = get_document(arg["_cls"])(
+                            pk=arg["_ref"].id)
+                    hydrated_references[arg_name] = reference_obj
                 elif arg_name == "id":
                     hydrated_references["id"] = from_global_id(args.pop("id", None))[1]
             args.update(hydrated_references)
@@ -286,7 +299,7 @@ def default_resolver(self, _root, info, required_fields=list(), **args):
                     if not info.context:
                         info.context = Context()
                     info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
-            elif _root is None:
+            elif _root is None or args:
                 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,
@@ -364,6 +377,9 @@ def chained_resolver(self, resolver, is_partial, root, info, **args):
                         if arg_name not in self.model._fields_ordered + ('first', 'last', 'before', 'after') + tuple(
                                 self.filter_args.keys()):
                             args_copy.pop(arg_name)
+                            if '.' in arg_name:
+                                operation = list(arg.keys())[0]
+                                args_copy[arg_name.replace('.', '__') + operation.replace('$', '__')] = arg[operation]
                     return self.default_resolver(root, info, required_fields, **args_copy)
                 else:
                     return resolved

graphene_mongo/tests/test_converter.py

@@ -305,8 +305,8 @@ class Meta:
         Article._fields["pub_date"], A._meta.registry
     )
     assert (
-        pubDate_field.kwargs["description"]
-        == "Publication Date\nThe date of first press."
+            pubDate_field.kwargs["description"]
+            == "Publication Date\nThe date of first press."
     )
 
     firstName_field = convert_mongoengine_field(