wip: Push for remote working

Author: abawchen

graphene_mongo/fields.py

@@ -63,10 +63,13 @@ def model(self):
 
     @property
     def args(self):
-        return to_arguments(
+        args = to_arguments(
             self._base_args or OrderedDict(),
             dict(self.field_args, **self.reference_args)
         )
+        print('args', args)
+        return args
+
 
     @args.setter
     def args(self, args):
@@ -92,24 +95,11 @@ 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()})
+                if isinstance(kv[1], Dynamic):
+                    r.update({kv[0]: self._field_args(node.fields.items())})
+                else:
+                    r.update({kv[0]: node.fields['id']._type.of_type()})
             return r
-        print('waw', self.fields.items())
-        # return {}
         return reduce(get_reference_field, self.fields.items(), {})
 
     @property
@@ -123,9 +113,8 @@ def get_query(cls, model, info, **args):
             return [], 0
 
         objs = model.objects()
-
+        print('hahaha', args)
         if args:
-            print('here!!!!')
             reference_fields = get_model_reference_fields(model)
             reference_args = {}
             for arg_name, arg in args.copy().items():
@@ -134,7 +123,6 @@ 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,7 +91,6 @@ class ProfessorVector(Document):
 
     meta = {'collection': 'test_professor_vector'}
     vec = ListField(FloatField())
-    last_name = StringField()
     metadata = EmbeddedDocumentField(ProfessorMetadata)
 
 

graphene_mongo/tests/test_relay_query.py

@@ -729,30 +729,29 @@ class Query(graphene.ObjectType):
     assert json.dumps(result.data, sort_keys=True) == json.dumps(
         expected, sort_keys=True)
 
-@pytest.mark.skip(reason="no way of currently testing this")
+# @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
-                }
+    query {
+      allProfessors(lastName: "5e06aa20-6805-4eef-a144-5615dedbe32b") {
+        edges {
+            node {
+                vec
             }
-          }
         }
+      }
+    }
     '''
     expected = {
         'allProfessors': {
             'edges': [
                 {
                     'node': {
-                        # 'id': 'abc',
                         'vec': [1.0, 2.3]
                     }
 
@@ -762,5 +761,6 @@ class Query(graphene.ObjectType):
     }
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
+    print(result.data)
     # assert not result.errors
     # assert dict(result.data['allProfessors']) == expected['allProfessors']