wip: Add FsFileType.data and its resolver

Author: abawchen

graphene_mongo/advanced_types.py

@@ -1,3 +1,4 @@
+import base64
 import mongoengine
 import graphene
 
@@ -8,6 +9,12 @@ def _resolve_type_coordinates(self, info):
     return self['coordinates']
 
 
+def _resolve_data(self, info):
+    v = getattr(self.instance, self.key)
+    data = v.read()
+    return base64.b64encode(data)
+
+
 class FsFile(mongoengine.Document):
 
     meta = {'collection': 'fs.files'}
@@ -22,6 +29,8 @@ class FsFileType(MongoengineObjectType):
     class Meta:
         model = FsFile
 
+    data = graphene.String(resolver=_resolve_data)
+
 
 class _TypeField(graphene.ObjectType):
 

graphene_mongo/tests/setup.py

@@ -31,11 +31,12 @@ def fixtures():
         editor1.avatar.put(f, content_type='image/jpeg')
     editor1.save()
 
-    """
     e = Editor.objects(first_name='Penny').first()
     avatar = e.avatar.read()
     print(e.avatar.content_type)
-    """
+    print(e.avatar.chunk_size)
+    print(e.avatar.md5)
+    print(type(avatar))
 
     editor2 = Editor(
         id='2',

graphene_mongo/tests/test_query.py

@@ -36,7 +36,8 @@ def resolve_editors(self, *args, **kwargs):
                     contentType,
                     chunkSize,
                     length,
-                    md5
+                    md5,
+                    chunk
                 }
             }
             editors {
@@ -53,7 +54,8 @@ def resolve_editors(self, *args, **kwargs):
                 'contentType': 'image/jpeg',
                 'chunkSize': 261120,
                 'length': 46928,
-                'md5': 'f3c657fd472fdc4bc2ca9056a1ae6106'
+                'md5': 'f3c657fd472fdc4bc2ca9056a1ae6106',
+                'chunk': 'abc'
             }
         },
         'editors': [{