如何使用 MongoEngine 从引用字段中检索 pdf/image？

Question

我发现使用 mongodb 中的烧瓶检索 class 引用的 pdf/image 文件有些困难。例如，我有这个模型：

class Users(db.Document):
    _id = db.StringField()
    name = db.StringField()
    picture = db.ReferenceField('fs.files')
    email = db.StringField()
    password = db.StringField()
    meta = {'collection': 'Users'}

用户 table 中记录的 JSON 如下所示：

{
    "_id": "1", 
    "name": "John Doe", 
    "picture": {
        "$ref": "fs.files", 
        "$id": {
            "$oid": "5e1...a932"
         }
     }, 
     "email":"john.doe@example.come", 
     "password": "12345"
}

在 Flask Restful api 中使用此模型，我正在尝试检索与用户关联的图像以在我的应用程序中显示。另外，添加新用户时，如何保存用户 table 中引用的文件？图像的参考存储在图片字段中。我也想以同样的方式对 pdf 执行此操作。

我试过查看 GridFS，但我不太了解它的工作原理或如何使用 mongoengine 在我的烧瓶 api 中实现它。谢谢

Answer 1

您可以使用 Flask 的 send_file 扩展来创建一个 url 加载静态文件作为响应。

from flask import send_file

@app.route('/get-image/<user>')
def get-image(user):
    """Serves static image loaded from db."""

    user = Users.objects(name=user).first()

    return send_file(io.BytesIO(user.picture.read()),
                     attachment_filename='image.jpg',
                     mimetype='image/jpg')

为了使上述解决方案生效，您应该在文档模型上使用 FileField() 而不是 ReferenceField()：

PS: 不知道能不能用ReferenceField来实现，下面的方法用的是gridfs，貌似比较合适

class Users(db.Document):
    _id = db.StringField()
    name = db.StringField()
    picture = db.FileField()
    email = db.StringField()
    password = db.StringField()
    meta = {'collection': 'Users'}

您可以像这样将文件加载到模型中：

user = Users.objects(name='User123').first()

with open('pic.jpg', 'rb') as fd:
    user.picture.put(fd, content_type = 'image/jpeg')
user.save()

希望它适合你

http://docs.mongoengine.org/guide/gridfs.html

如何使用 MongoEngine 从引用字段中检索 pdf/image？

How to retrieve a pdf/image from reference field using MongoEngine?

python

mongodb

mongoengine

flask-restful

flask-mongoengine