PyMongo使用集合查询结果构建删除查询Cursor类型无法拼接报错求助

错误原因说明 #

1. 第一次报错是因为tempCollection.find()返回的是Cursor游标对象，不是字符串或列表，无法直接和字符串拼接
2. 手动拼接Mongo Shell格式的查询字符串无效，是因为PyMongo的查询参数要求传入原生Python字典，而非Mongo Shell可用的字符串格式，驱动会自动处理Python数据类型到MongoDB类型的映射
3. 原有代码中print(x.deleted_count + "records deleted.")还隐含类型错误：deleted_count是整数类型，无法直接和字符串拼接

正确实现代码 #

简洁写法（推荐，数据量不大时使用） #

直接用distinct方法获取所有_id的列表，无需手动遍历游标：

from bson.objectid import ObjectId

def remove_from_source():
    # 直接提取tempCollection中所有_id的值，返回ObjectId类型的列表
    id_list = tempCollection.distinct("_id")
    # 构造符合PyMongo规范的查询字典
    delete_query = {"_id": {"$in": id_list}}
    # 执行批量删除
    result = sourceCollection.delete_many(delete_query)
    # 格式化输出删除数量
    print(f"{result.deleted_count} records deleted.")

自定义遍历写法（适合需要对_id做额外处理的场景） #

如果你需要在获取_id的过程中做过滤等操作，可以手动遍历游标：

def remove_from_source():
    id_cursor = tempCollection.find({}, {"_id": 1})
    # 遍历游标提取所有_id，生成列表
    id_list = [doc["_id"] for doc in id_cursor]
    delete_query = {"_id": {"$in": id_list}}
    result = sourceCollection.delete_many(delete_query)
    print(f"{result.deleted_count} records deleted.")

额外说明 #

• 如果待处理的_id量级超过10万，建议分批处理避免占用过多内存
• 所有PyMongo的操作参数都不需要手动拼接Mongo Shell格式的字符串，直接用Python的字典、列表、字符串、ObjectId等原生类型即可

内容的提问来源于stack exchange，提问作者Ben Fourie