Merge branch 'support-generic-reference-field-in-args'

Author: arunsureshkumar

graphene_mongo/fields.py

@@ -220,8 +220,12 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                     hydrated_references[arg_name] = reference_obj
                 elif arg_name in self.model._fields_ordered and isinstance(getattr(self.model, arg_name),
                                                                            mongoengine.fields.GenericReferenceField):
-                    reference_obj = get_document(self.registry._registry_string_map[from_global_id(arg)[0]])(
-                        pk=from_global_id(arg)[1])
+                    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 '__near' in arg_name and isinstance(getattr(self.model, arg_name.split('__')[0]),
                                                          mongoengine.fields.PointField):