以下是一个示例代码,用于保存文档到特定的集合。如果集合不存在,则报错。
from pymongo import MongoClient
def save_to_collection(collection_name, document):
# 连接 MongoDB
client = MongoClient('mongodb://localhost:27017/')
# 指定数据库
db = client['mydatabase']
# 检查集合是否存在
if collection_name not in db.list_collection_names():
raise ValueError('集合不存在!')
# 获取集合
collection = db[collection_name]
# 插入文档
collection.insert_one(document)
# 关闭连接
client.close()
# 测试代码
try:
document = {'title': '文档标题', 'content': '文档内容'}
save_to_collection('mycollection', document)
print('文档保存成功!')
except ValueError as e:
print(e)
在上面的示例中,我们使用了pymongo库来连接 MongoDB 数据库,并使用了MongoClient类来建立连接。然后,我们指定了要使用的数据库('mydatabase')并检查了集合是否存在。如果集合不存在,我们抛出了一个ValueError异常。如果集合存在,我们获取该集合并使用insert_one()方法将文档插入到集合中。
请注意,在使用完数据库后,我们需要调用close()方法来关闭连接,以释放资源。
在测试代码中,我们创建了一个示例文档,并调用save_to_collection()函数来保存文档到集合中。如果集合不存在,将会抛出异常并打印错误信息。如果保存成功,将打印"文档保存成功!"。
你可以根据自己的需求修改数据库连接信息、数据库名称、集合名称和文档内容。
上一篇:保存文档