Removed unwanted conditions

Author: arunsureshkumar

graphene_mongo/advanced_types.py

@@ -3,7 +3,6 @@
 
 
 class FileFieldType(graphene.ObjectType):
-
     content_type = graphene.String()
     md5 = graphene.String()
     chunk_size = graphene.Int()
@@ -36,7 +35,6 @@ def resolve_data(self, info):
 
 
 class _CoordinatesTypeField(graphene.ObjectType):
-
     type = graphene.String()
 
     def resolve_type(self, info):
@@ -47,17 +45,19 @@ def resolve_coordinates(self, info):
 
 
 class PointFieldType(_CoordinatesTypeField):
-
     coordinates = graphene.List(graphene.Float)
 
 
-class PolygonFieldType(_CoordinatesTypeField):
+class PointFieldInputType(graphene.InputObjectType):
+    type = graphene.String(default_value="Point")
+    coordinates = graphene.List(graphene.Float, required=True)
 
+
+class PolygonFieldType(_CoordinatesTypeField):
     coordinates = graphene.List(graphene.List(graphene.List(graphene.Float)))
 
 
 class MultiPolygonFieldType(_CoordinatesTypeField):
-
     coordinates = graphene.List(
         graphene.List(graphene.List(graphene.List(graphene.Float)))
     )

graphene_mongo/fields.py

@@ -24,7 +24,7 @@
     FileFieldType,
     PointFieldType,
     MultiPolygonFieldType,
-    PolygonFieldType,
+    PolygonFieldType, PointFieldInputType,
 )
 from .converter import convert_mongoengine_field, MongoEngineConversionError
 from .registry import get_global_registry
@@ -113,6 +113,7 @@ def is_filterable(k):
                     converted.type(),
                     (
                             FileFieldType,
+                            PointFieldType,
                             MultiPolygonFieldType,
                             graphene.Union,
                             PolygonFieldType,
@@ -172,12 +173,18 @@ def reference_args(self):
         def get_reference_field(r, kv):
             field = kv[1]
             mongo_field = getattr(self.model, kv[0], None)
+            if isinstance(mongo_field, mongoengine.PointField):
+                r.update({kv[0]: graphene.Argument(PointFieldInputType)})
+                return r
             if isinstance(
                     mongo_field,
                     (mongoengine.LazyReferenceField, mongoengine.ReferenceField, mongoengine.GenericReferenceField),
             ):
                 r.update({kv[0]: graphene.ID()})
                 return r
+            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:
@@ -217,6 +224,8 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                                                          mongoengine.fields.PointField):
                     location = args.pop(arg_name, None)
                     hydrated_references[arg_name] = location["coordinates"]
+                    if (arg_name.split('__')[0] + "__max_distance") not in args:
+                        hydrated_references[arg_name.split('__')[0] + "__max_distance"] = 10000
                 elif arg_name == "id":
                     hydrated_references["id"] = from_global_id(args.pop("id", None))[1]
             args.update(hydrated_references)
@@ -229,6 +238,7 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                 args.update(queryset_or_filters)
         if limit is not None:
             if reversed:
+                order_by = ""
                 if self.order_by:
                     order_by = self.order_by + ",-pk"
                 else: