Support added for PointFieldType in agrs

Support added for max_distance filter in PointFieldType

Author: arunsureshkumar

graphene_mongo/fields.py

@@ -149,10 +149,13 @@ def filter_args(self):
         if self._type._meta.filter_fields:
             for field, filter_collection in self._type._meta.filter_fields.items():
                 for each in filter_collection:
-                    filter_type = getattr(
-                        graphene,
-                        str(self._type._meta.fields[field].type).replace("!", ""),
-                    )
+                    if each == 'max_distance' and str(self._type._meta.fields[field].type) == 'PointFieldType':
+                        filter_type = graphene.Int
+                    else:
+                        filter_type = getattr(
+                            graphene,
+                            str(self._type._meta.fields[field].type).replace("!", ""),
+                        )
 
                     # handle special cases
                     advanced_filter_types = {
@@ -175,10 +178,8 @@ def get_reference_field(r, kv):
             mongo_field = getattr(self.model, kv[0], None)
             if isinstance(
                     mongo_field,
-                    (mongoengine.LazyReferenceField, mongoengine.ReferenceField),
+                    (mongoengine.LazyReferenceField, mongoengine.ReferenceField, mongoengine.GenericReferenceField),
             ):
-                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)):
@@ -216,6 +217,10 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                     reference_obj = get_document(self.registry._registry_string_map[from_global_id(arg)[0]])(
                         pk=from_global_id(arg)[1])
                     hydrated_references[arg_name] = reference_obj
+                elif '__near' in arg_name and isinstance(getattr(self.model, arg_name.split('__')[0]),
+                                                         mongoengine.fields.PointField):
+                    location = args.pop(arg_name, None)
+                    hydrated_references[arg_name] = location["coordinates"]
                 elif arg_name == "id":
                     hydrated_references["id"] = from_global_id(args.pop("id", None))[1]
             args.update(hydrated_references)

graphene_mongo/types.py

@@ -41,9 +41,9 @@ def construct_fields(model, registry, only_fields, exclude_fields):
             # Take care of list of self-reference.
             document_type_obj = field.field.__dict__.get("document_type_obj", None)
             if (
-                document_type_obj == model._class_name
-                or isinstance(document_type_obj, model)
-                or document_type_obj == model
+                    document_type_obj == model._class_name
+                    or isinstance(document_type_obj, model)
+                    or document_type_obj == model
             ):
                 self_referenced[name] = field
                 continue
@@ -67,7 +67,6 @@ def construct_self_referenced_fields(self_referenced, registry):
 
 
 class MongoengineObjectTypeOptions(ObjectTypeOptions):
-
     model = None
     registry = None  # type: Registry
     connection = None
@@ -78,22 +77,22 @@ class MongoengineObjectTypeOptions(ObjectTypeOptions):
 class MongoengineObjectType(ObjectType):
     @classmethod
     def __init_subclass_with_meta__(
-        cls,
-        model=None,
-        registry=None,
-        skip_registry=False,
-        only_fields=(),
-        required_fields=(),
-        exclude_fields=(),
-        filter_fields=None,
-        connection=None,
-        connection_class=None,
-        use_connection=None,
-        connection_field_class=None,
-        interfaces=(),
-        _meta=None,
-        order_by=None,
-        **options
+            cls,
+            model=None,
+            registry=None,
+            skip_registry=False,
+            only_fields=(),
+            required_fields=(),
+            exclude_fields=(),
+            filter_fields=None,
+            connection=None,
+            connection_class=None,
+            use_connection=None,
+            connection_field_class=None,
+            interfaces=(),
+            _meta=None,
+            order_by=None,
+            **options
     ):
 
         assert is_valid_mongoengine_model(model), (