Merge pull request #5 from abawchen/feat-default-filter

Feat: Default filter behavior

Author: abawchen

graphene_mongo/fields.py

@@ -1,11 +1,13 @@
 from __future__ import absolute_import
 
-from functools import partial
+from collections import OrderedDict
+from functools import partial, reduce
 
 from graphene import Field
 from graphene.relay import ConnectionField
 from graphene.relay.connection import PageInfo
 from graphql_relay.connection.arrayconnection import connection_from_list_slice
+from graphene.types.argument import to_arguments
 
 from .utils import maybe_queryset
 
@@ -52,8 +54,36 @@ def node_type(self):
     def model(self):
         return self.node_type._meta.model
 
+    @property
+    def args(self):
+        return to_arguments(
+            self._base_args or OrderedDict(), self.default_filter_args
+        )
+
+    @args.setter
+    def args(self, args):
+        self._base_args = args
+
+    @property
+    def default_filter_args(self):
+        return reduce(
+            lambda r, kv: r.update({kv[0]: kv[1]._type._of_type()}) or r if hasattr(kv[1], '_type') else r,
+            self.fields.items(),
+            {}
+        )
+
+    @property
+    def filter_fields(self):
+        return self._type._meta.filter_fields
+
+    @property
+    def fields(self):
+        return self._type._meta.fields
+
     @classmethod
     def get_query(cls, model, info, **args):
+        if args:
+            return model.objects().filter(**args)
         return model.objects()
 
     @classmethod
@@ -84,6 +114,5 @@ def connection_resolver(cls, resolver, connection, model, root, info, **args):
         return connection
 
     def get_resolver(self, parent_resolver):
-        ('??')
         return partial(self.connection_resolver, parent_resolver, self.type, self.model)
 

graphene_mongo/tests/test_query.py

@@ -292,9 +292,46 @@ class Mutation(graphene.ObjectType):
     assert not result.errors
     assert result.data == expected
 
-# TODO:
+@with_local_registry
 def test_should_filter():
-    pass
+    class ArticleNode(MongoengineObjectType):
+
+        class Meta:
+            model = Article
+            interfaces = (Node,)
+            filter_fields = ('headline',)
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        articles = MongoengineConnectionField(ArticleNode)
+
+    query = '''
+        query ArticleQuery {
+          articles(headline: "World") {
+            edges {
+                node {
+                    headline
+                }
+            }
+          }
+        }
+    '''
+    expected = {
+        'articles': {
+            'edges': [
+                {
+                    'node': {
+                        'headline': 'World'
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert result.data == expected
+
 
 # TODO:
 def test_should_paging():

graphene_mongo/types.py

@@ -33,7 +33,7 @@ class MongoengineObjectTypeOptions(ObjectTypeOptions):
     model = None  # type: Model
     registry = None  # type: Registry
     connection = None  # type: Type[Connection]
-
+    filter_fields = ()
 
 class MongoengineObjectType(ObjectType):
 
@@ -78,6 +78,7 @@ def __init_subclass_with_meta__(cls, model=None, registry=None, skip_registry=Fa
         _meta.model = model
         _meta.registry = registry
         _meta.fields = mongoengine_fields
+        _meta.filter_fields = filter_fields
         _meta.connection = connection
         super(MongoengineObjectType, cls).__init_subclass_with_meta__(
             _meta=_meta, interfaces=interfaces, **options