如何使用pymongo将集合转储到json文件

Question

我正在尝试将收藏转储到。json文件，但在看过pymongo教程后，我找不到任何与之相关的东西。

教程链接：https://api.mongodb.com/python/current/tutorial.html

Answer 1

接受的解决方案生成无效的JSON。它的结果是在右方括号]之前尾随逗号，。JSON规范不允许尾随逗号。请参阅此答案和此参考资料。

为了基于公认的解决方案，我使用了以下方法：

from bson.json_util import dumps
from pymongo import MongoClient
import json

if __name__ == '__main__':
    client = MongoClient()
    db = client.db_name
    collection = db.collection_name
    cursor = collection.find({})
    with open('collection.json', 'w') as file:
        json.dump(json.loads(dumps(cursor)), file)

Answer 2

下面是另一种不在结束方括号前保存，的方法。同时使用打开的来节省一些空间。

filter = {"type": "something"}
type_documents = db['cluster'].find(filter)
type_documents_count = db['cluster'].count_documents(filter)

with open("type_documents.json", "w") as file:
    file.write('[')
    # Start from one as type_documents_count also starts from 1.
    for i, document in enumerate(type_documents, 1):
        file.write(json.dumps(document, default=str))
        if i != type_documents_count:
            file.write(',')
    file.write(']')

如果迭代次数等于文档数（它保存的最后一个文档），它基本上不会写逗号。

Answer 3

只需获取所有文档并将其保存到文件中，例如：

from bson.json_util import dumps
from pymongo import MongoClient

if __name__ == '__main__':
    client = MongoClient()
    db = client.db_name
    collection = db.collection_name
    cursor = collection.find({})
    with open('collection.json', 'w') as file:
        file.write('[')
        for document in cursor:
            file.write(dumps(document))
            file.write(',')
        file.write(']')