from pymongo import MongoClient
import pandas as pd


# 连接到MongoDB（这里假设MongoDB运行在本地默认端口上）
mgclient = MongoClient('localhost', 27017)

# 选择数据库（如果你没有指定数据库，MongoDB会使用默认的'test'数据库）
db = mgclient['test']  # 替换'your_database_name'为你的数据库名
test1_collection=db["test1"].find({})
test1_list=list(test1_collection)
test2_collection=db["test2"].find({})
test2_list=list(test2_collection)

# 这种方式只适合插入存在的数据。
for document1 in test1_list:
    document1.pop("_id")
    document1["codename"]=document1["code"]
    # document1["codename"] = document1["code"]
    for document2 in test2_list:
        # dataframe包含关系不是很好merge
        if document1["code"]==document2["code"] and document1["code"].startswith(document2["code"]):
            document1["codename"]=document2["name"]
            print(document1)
            # 如果只用写入能够匹配上的，在这里insert
            # db["test3"].insert_one(document1)
    db["test3"].insert_one(document1)