Question

使用insert_many时，使用pymongo持久保存文档时遇到一些问题。

我正在将字典列表移交给insert_many，它在执行插入操作的同一脚本中正常工作。脚本完成后就更少了。

def row_to_doc(row):
    rowdict = row.to_dict()
    for key in rowdict:
        val = rowdict[key]
        if type(val) == float or type(val) == np.float64:
            if np.isnan(val):
                # If we want a SQL style document collection
                rowdict[key] = None
                # If we want a NoSQL style document collection
                # del rowdict[key]
    return rowdict

def dataframe_to_collection(df):
    n = len(df)
    doc_list = []
    for k in range(n):
        doc_list.append(row_to_doc(df.iloc[k]))
    return doc_list

def get_mongodb_client(host="localhost", port=27017):
    return MongoClient(host, port)

def create_collection(client):
    db = client["material"]
    return db["master-data"]

def add_docs_to_mongo(collection, doc_list):
    collection.insert_many(doc_list)

def main():
    client = get_mongodb_client()
    csv_fname = "some_csv_fname.csv"
    df = get_clean_csv(csv_fname)
    doc_list = dataframe_to_collection(df)

    collection = create_collection(client)
    add_docs_to_mongo(collection, doc_list)

    test_doc = collection.find_one({"MATERIAL": "000000000000000001"})

当我打开另一个python REPL并开始使用client.material.master_data或collection.find_one({"MATERIAL": "000000000000000001"})浏览collection.count_documents({})集合时，find_one的输入为None，而{{1 }}。

是否存在需要调用某种方法将数据持久保存到磁盘的步骤？ mongo客户端API中的count_documents听起来很像，但这只是从我已阅读的内容中插入文档的另一种方式。任何帮助将不胜感激。

Answer 1

问题是我正在通过client.db_name.collection_name获取我的收藏集，而没有获得与我用代码创建的收藏集相同的收藏集。 client.db_name["collection-name"]解决了我的问题。很奇怪。

pymongo中的insert_many不持久

1 个答案: