Python保存MongoDB上的文件到本地的方法

下面是Python保存MongoDB上的文件到本地的方法的完整实例教程，包含两个示例：

环境配置

首先需要安装pymongo和gridfs库，可以使用pip进行安装：

pip install pymongo
pip install gridfs

示例1：保存一张图片文件

这个示例展示了如何从MongoDB中读取一个图片文件，并将其保存到本地。首先，假设MongoDB中已经存储了一张名为“image.jpg”的图片。

import pymongo
from gridfs import GridFS
import io

# 连接MongoDB
client = pymongo.MongoClient()
db = client.test
fs = GridFS(db)

# 读取图片
cursor = fs.find({'filename': 'image.jpg'})
if cursor.count() > 0:
    # 保存到本地
    with io.open('image.jpg', 'wb') as f:
        f.write(cursor.next().read())

代码解析：

• 使用pymongo连接MongoDB，并获取GridFS实例
• 使用GridFS的find()方法查找名为“image.jpg”的文件，如果存在则进行保存
• 由于MongoDB中的文件以二进制的形式存储，因此需要使用io库将其写入到本地文件系统中

示例2：保存一个PDF文件

这个示例展示了如何从MongoDB中读取一个PDF文件，并将其保存到本地。同样假设MongoDB中已经存储了一个名为“document.pdf”的文件。

import pymongo
from gridfs import GridFS
import io

# 连接MongoDB
client = pymongo.MongoClient()
db = client.test
fs = GridFS(db)

# 读取PDF文件
cursor = fs.find({'filename': 'document.pdf'})
if cursor.count() > 0:
    # 保存到本地
    with io.open('document.pdf', 'wb') as f:
        f.write(cursor.next().read())

代码解析：

和第一个示例一样，只是文件类型不同，但是方法是一样的。

有了这两个示例，你就可以将MongoDB中的文件保存到本地了。如果需要保存其他类型的文件，只需要将代码稍微修改一下即可。