MongoDB：带有数组过滤器的UPSERT

Question

我有这样的收藏：

mc.db.collection.insert_many([
    {"key_array": [1], "another_array": ["a"]},
    {"key_array": [2, 3], "another_array": ["b"]},
    {"key_array": [4], "another_array": ["c", "d"]},    
])

我正在使用这种更新：

mc.db.collection.update_one(
    {"key_array": 5},
    {"$addToSet": {"another_array": "f"}},
    upsert=True
)

它可以在更新中效果很好，但是在尝试上升时我遇到了麻烦： 它创建一个使用非阵列 key_array 字段的文档，

{
    "_id": ObjectId(...)
    "key_array": 5,
    "another_array": ["f"]
}

而我也想拥有一个文档

{
    "_id": ObjectId(...)
    "key_array": [5],
    "another_array": ["f"]
}

，我也不能使用 {“ key_array”：[5]} 样式查询，因为它与现有数组与长度＆gt; 1。

那么，是否有机会在更新上保存此类行为，并在插入列表上接收正确的文档结构？

任何帮助将不胜感激

Answer 1

这应该有帮助。

mc.db.collection.update_one(
    {"key_array": 5},
    {
        "$addToSet": {"another_array": "f"},
        "$setOnInsert": {"key_array": [5], ...}
    },
    upsert=True
)

Answer 2

怎么样。

db.collection.update({
  "key_array": 5
},
{
  "$addToSet": {
    "another_array": "f",
    
  },
  "$set": {
    "key_array": [
      5
    ]
  }
},
{
  "upsert": true
})

https://mongoplayground.net/p/4ydhkuzr2i6

Answer 3

好的，最后，我已经通过两个连续的更新解决了这个问题，该问题是问题中指定的第一个更新 - 具有非阵列查询字段的UPSERTS，第二个将字段转换为数组（如果属于其他类型）。

from pymongo import UpdateOne

bulk = [
    UpdateOne(
        {"key_array": 6},
        {"$addToSet": {"another_array": "e"}},
        upsert=True
    ),
    UpdateOne(
        {
            "$and": [{"key_array": 6},
                     {"key_array": {"$not": {"$type": "array"}}}]
        },
        [{"$set": { "key_array": ["$key_array"]}}]
    )
]

mc.db.collection.bulk_write(bulk)

但是我仍在寻找一种更优雅的方法来做这种事情。