wip: Add fields._field_args first

Author: abawchen

graphene_mongo/fields.py

@@ -72,8 +72,7 @@ def args(self):
     def args(self, args):
         self._base_args = args
 
-    @property
-    def field_args(self):
+    def _field_args(self, items):
         def is_filterable(v):
             return not isinstance(v, (ConnectionField, Dynamic))
 
@@ -82,16 +81,35 @@ def get_type(v):
                 return v.type.of_type()
             return v.type()
 
-        return {k: get_type(v) for k, v in self.fields.items()
-                if is_filterable(v)}
+        return {k: get_type(v) for k, v in items if is_filterable(v)}
+
+    @property
+    def field_args(self):
+        return self._field_args(self.fields.items())
 
     @property
     def reference_args(self):
         def get_reference_field(r, kv):
             if callable(getattr(kv[1], 'get_type', None)):
                 node = kv[1].get_type()._type._meta
+                # print('abaw', kv[0], node)
+                # print(node.__dict__)
+                # print(node.fields['id'])
+                # print(type(node.fields['id']))
+                # print(node.fields['id']._type)
+                # print(node.fields['id']._type.of_type())
+                # print('?')
                 r.update({kv[0]: node.fields['id']._type.of_type()})
+                """
+                print('a', node.__dict__)
+                print('b', kv[1])
+                print('c', node)
+                print('d', node.__dict__)
+                """
+                # r.update({kv[0]: kv[1]._type.of_type()})
             return r
+        print('waw', self.fields.items())
+        # return {}
         return reduce(get_reference_field, self.fields.items(), {})
 
     @property
@@ -107,6 +125,7 @@ def get_query(cls, model, info, **args):
         objs = model.objects()
 
         if args:
+            print('here!!!!')
             reference_fields = get_model_reference_fields(model)
             reference_args = {}
             for arg_name, arg in args.copy().items():
@@ -115,6 +134,7 @@ def get_query(cls, model, info, **args):
                     pk = from_global_id(args.pop(arg_name))[-1]
                     reference_obj = reference_model.document_type_obj.objects(pk=pk).get()
                     reference_args[arg_name] = reference_obj
+            print('reference_args', reference_args)
 
             args.update(reference_args)
             first = args.pop('first', None)

graphene_mongo/tests/models.py

@@ -91,6 +91,7 @@ class ProfessorVector(Document):
 
     meta = {'collection': 'test_professor_vector'}
     vec = ListField(FloatField())
+    last_name = StringField()
     metadata = EmbeddedDocumentField(ProfessorMetadata)
 
 

graphene_mongo/tests/test_relay_query.py

@@ -1,4 +1,5 @@
 import json
+import pytest
 
 import graphene
 
@@ -11,7 +12,8 @@
                     PlayerNode,
                     ReporterNode,
                     ChildNode,
-                    ParentWithRelationshipNode)
+                    ParentWithRelationshipNode,
+                    ProfessorVectorNode,)
 from ..fields import MongoengineConnectionField
 
 
@@ -726,3 +728,39 @@ class Query(graphene.ObjectType):
     assert not result.errors
     assert json.dumps(result.data, sort_keys=True) == json.dumps(
         expected, sort_keys=True)
+
+@pytest.mark.skip(reason="no way of currently testing this")
+def test_should_query_with_embedded_document(fixtures):
+
+    class Query(graphene.ObjectType):
+
+        all_professors = MongoengineConnectionField(ProfessorVectorNode)
+
+    query = '''
+        query EditorQuery {
+          allProfessors() {
+            edges {
+                node {
+                    vec
+                }
+            }
+          }
+        }
+    '''
+    expected = {
+        'allProfessors': {
+            'edges': [
+                {
+                    'node': {
+                        # 'id': 'abc',
+                        'vec': [1.0, 2.3]
+                    }
+
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    # assert not result.errors
+    # assert dict(result.data['allProfessors']) == expected['allProfessors']

graphene_mongo/tests/types.py

@@ -106,24 +106,35 @@ class Meta:
 
 
 class ChildNode(MongoengineObjectType):
+
     class Meta:
         model = Child
         interfaces = (Node,)
 
 
 class ChildRegisteredBeforeNode(MongoengineObjectType):
+
     class Meta:
         model = ChildRegisteredBefore
         interfaces = (Node, )
 
 
 class ParentWithRelationshipNode(MongoengineObjectType):
+
     class Meta:
         model = ParentWithRelationship
         interfaces = (Node, )
 
 
 class ChildRegisteredAfterNode(MongoengineObjectType):
+
     class Meta:
         model = ChildRegisteredAfter
         interfaces = (Node, )
+
+
+class ProfessorVectorNode(MongoengineObjectType):
+
+    class Meta:
+        model = ProfessorVector
+        interfaces = (Node, )