如何在Python中修改GridFS默认存储桶名称

嗨，这个问题我之前也碰到过，其实修改默认的存储桶名称超简单，只要在初始化GridFS的时候多传一个参数就行，下面给你一步步说清楚怎么操作：

1. 导入必要的模块 #

首先得确保你已经安装了pymongo，然后导入需要的类：

from pymongo import MongoClient
from gridfs import GridFS

2. 连接MongoDB并获取数据库 #

先建立和MongoDB的连接，拿到对应的数据库对象：

# 连接本地MongoDB，根据你的实际地址修改
client = MongoClient('mongodb://localhost:27017/')
# 替换成你自己的数据库名
db = client['your_database_name']

3. 初始化GridFS时指定自定义存储桶名称 #

关键就在这一步！初始化GridFS的时候，通过bucket_name参数设置你想要的名称（比如这里的"custom"）：

# 指定自定义存储桶为"custom"
fs = GridFS(db, bucket_name="custom")

4. 验证自定义桶的使用 #

接下来你就可以像平时用GridFS一样上传、读取文件了，这些操作都会作用在custom.chunks和custom.files集合上：

• 上传文件示例：

# 上传一个本地文件到自定义桶
with open("example.txt", "rb") as f:
    file_id = fs.put(f, filename="example.txt")

• 读取文件示例：

# 从自定义桶里读取文件
file_data = fs.get(file_id).read()
print(file_data.decode('utf-8'))

额外说明 #

如果你之后要查看或者操作这两个自定义集合，直接通过数据库对象访问就行：

# 获取custom.files集合
custom_files_col = db['custom.files']
# 获取custom.chunks集合
custom_chunks_col = db['custom.chunks']

这样操作之后，GridFS就不会再使用默认的fs前缀集合了，完全按照你指定的"custom"来创建和使用对应的两个集合~

内容的提问来源于stack exchange，提问作者Zakir Ayub