Support for objects.order_by

Use it like this:
class XXX(MongoengineObjectType):
    class Meta:
        model:SomeModel
        interfaces:(Node,)
        order_by: 'some order_by clause used by
                   SomeeModel.object().order_by'

That's all, try it out.

Signed-off-by: Zhang Yuzheng <[email protected]>

Author: Zhang Yuzheng

graphene_mongo/fields.py

@@ -50,6 +50,10 @@ def node_type(self):
     def model(self):
         return self.node_type._meta.model
 
+    @property
+    def order_by(self):
+        return self.node_type._meta.order_by
+
     @property
     def registry(self):
         return getattr(self.node_type._meta, "registry", get_global_registry())
@@ -182,7 +186,7 @@ def get_queryset(self, model, info, **args):
                 return queryset_or_filters
             else:
                 args.update(queryset_or_filters)
-        return model.objects(**args)
+        return model.objects(**args).order_by(self.order_by)
 
     def default_resolver(self, _root, info, **args):
         args = args or {}

graphene_mongo/types.py

@@ -71,6 +71,7 @@ class MongoengineObjectTypeOptions(ObjectTypeOptions):
     registry = None  # type: Registry
     connection = None
     filter_fields = ()
+    order_by = None
 
 
 class MongoengineObjectType(ObjectType):
@@ -89,6 +90,7 @@ def __init_subclass_with_meta__(
         connection_field_class=None,
         interfaces=(),
         _meta=None,
+        order_by=None,
         **options
     ):
 
@@ -155,6 +157,7 @@ def __init_subclass_with_meta__(
         # Save them for later
         _meta.only_fields = only_fields
         _meta.exclude_fields = exclude_fields
+        _meta.order_by = order_by
 
         super(MongoengineObjectType, cls).__init_subclass_with_meta__(
             _meta=_meta, interfaces=interfaces, **options