mongoengine教程(4)——文件存储

MongoDB的GridFS支持直接在数据库中存储文件。要在MongoEngine中使用GridFS,只要使用 FileField 对象即可。以下是一个例子:

class Animal(Document):
    genus = StringField()
    family = StringField()
    photo = FileField()

marmot = Animal(genus='Marmota', family='Sciuridae')

marmot_photo = open('gtk.png', 'rb')
marmot.photo.put(marmot_photo, content_type = 'image/png')
marmot.save()

这个例子将 gtk.png 这个图片存入了数据库中。
文件的读取也很简单:

marmot = Animal.objects(genus='Marmota').first()
photo = marmot.photo.read()
content_type = marmot.photo.content_type

FileField不仅可以存储文件,还可以用来存储数据流。只是操作上略微不同。

要存储数据流,首先先创建一个新的文件,然后再往里面写入数据。

marmot.photo.new_file()
marmot.photo.write('some_image_data')
marmot.photo.write('some_more_image_data')
marmot.photo.close()
marmot.save()

如果要删除存储在数据库中的文件,只需要调用该文件对象的 delete 方法:

marmot.photo.delete()

注意:一条文档记录中的FileField字段只保存了对GridFS集合中该文件的ID引用。这意味着如果该文档被删除了,对用的文件不会被删除。因此在删除这类文档时需要小心,以免出现孤立文件。

对于已存储的文件可以进行替换修改:

another_marmot = open('python.png', 'rb')
marmot.photo.replace(another_marmot, content_type='image/png')