Merge branch 'master' into support-monogengine-field-enum

Author: arunsureshkumar

graphene_mongo/converter.py

@@ -88,6 +88,7 @@ def convert_point_to_field(field, registry=None):
                           required=field.required)
 
 
+
 @convert_mongoengine_field.register(mongoengine.PolygonField)
 def convert_polygon_to_field(field, registry=None):
     return graphene.Field(advanced_types.PolygonFieldType, description=get_field_description(field, registry),
@@ -106,6 +107,7 @@ def convert_file_to_field(field, registry=None):
                           required=field.required)
 
 
+
 @convert_mongoengine_field.register(mongoengine.ListField)
 @convert_mongoengine_field.register(mongoengine.EmbeddedDocumentListField)
 def convert_field_to_list(field, registry=None):

graphene_mongo/fields.py

@@ -397,9 +397,17 @@ def chained_resolver(self, resolver, is_partial, root, info, **args):
                                 args_copy['pk__in'] = arg['$in']
                             elif "$ne" in arg:
                                 args_copy['pk__ne'] = arg['$ne']
+                                operation = list(arg.keys())[0]
+                                args_copy['pk' + operation.replace('$', '__')] = arg[operation]
                             if '.' in arg_name:
                                 operation = list(arg.keys())[0]
                                 args_copy[arg_name.replace('.', '__') + operation.replace('$', '__')] = arg[operation]
+                        else:
+                            operations = ["$lte", "$gte", "$ne", "$in"]
+                            if isinstance(arg, dict) and any(op in arg for op in operations):
+                                operation = list(arg.keys())[0]
+                                args_copy[arg_name + operation.replace('$', '__')] = arg[operation]
+                                del args_copy[arg_name]
                     return self.default_resolver(root, info, required_fields, **args_copy)
                 else:
                     return resolved