是否有一种更简单的方法可以使用Python在MongoDB中更新整个记录？

Question

我的用例是：我将JSON有效载荷传递到Python程序中，该程序将记录列入我的MongoDB集合。它检查集合记录中的唯一组合，并在“时间戳”更大的情况下更新记录，如果找不到组合并保留“ Timestamp”较低的组合，则UPSERTS。

payload = {'stCode': 'ABC1', 'skCode': 'CDE2', 'batchCode': 'FGH3', ts: '2022/07/04 09:48:36'}
inp_ts = payload['ts']
inp_ts = datetime.strptime(inp_ts, '%Y/%m/%d %H:%M:%S')
inp_ts_iso = inp_ts.isoformat()

stCode = payload['stCode']
skCode = payload['skCode']
batchCode = payload['batchCode']
    
result = collection.find_one({'stCode': stCode,'skCode': skCode, 'batchCode': batchCode }) 

if result is None:
    collection.insert_one(payload) #Condition1: Insert record if not present
        
else:
    result_ts = result['ts']
    result_ts = datetime.strptime(result_ts, '%Y/%m/%d %H:%M:%S')
    result_ts_iso = result_ts.isoformat() #since the date is stored as string in the database
        
    if inp_ts_iso > result_ts_iso:
        temps={"storeCode": storeCode,"skuCode": skuCode, "batchCode": batchCode }, {"$set": payload}
        collection.update_one(*temps) #Condition2: Update record if timestamp is older
        print('Records successfully written to MongoDB')
            
    else:
        print("no records written") #Condition3: Do nothing if timestamp is newer

我知道这不是代码中最漂亮的，但我希望它更简单和简短。是否有一种方法可以通过没有Lambda进行逻辑测试而实现同样的方法，而是在Mongo的一边进行？我已经看到了复杂的update_one查询，但我无法为自己的利益而工作。提前致谢！

Answer 1

是的，您可以，您可以以几种不同的方式进行操作，但是它们都涉及管道更新，您可以在更新中使用聚合操作员。

我认为这是实现这一目标的最优雅方法：（底部的方法摘要）

const payload = {'stCode': 'ABC1', 'skCode': 'CDE2', 'batchCode': 'FGH3', ts: '2022/07/04 09:48:36'}
const stCode = payload['stCode']
const skCode = payload['skCode']
const batchCode = payload['batchCode']

db.collection.updateOne(
    {'stCode': stCode,'skCode': skCode, 'batchCode': batchCode },
    [
      {
        "$replaceRoot": {
          "newRoot": {
            "$mergeObjects": [
              "$ROOT",
              {
                $cond: [
                  {
                    $gt: [
                      "$ts",
                      payload.ts
                    ]
                  },
                  {},
                  payload
                ]
              }
            ]
          }
        }
      }
    ],
    {
      "upsert": true
    }
)

或pymongo：

payload = {'stCode': 'ABC1', 'skCode': 'CDE2', 'batchCode': 'FGH3', ts: '2022/07/04 09:48:36'}
stCode = payload['stCode']
skCode = payload['skCode']
batchCode = payload['batchCode']

collection.update_one(
    {'stCode': stCode, 'skCode': skCode, 'batchCode': batchCode},
    [
        {
            "$replaceRoot": {
                "newRoot": {
                    "$mergeObjects": [
                        "$ROOT",
                        {
                            "$cond": [
                                {
                                    "$gt": [
                                        "$ts",
                                        payload['ts']
                                    ]
                                },
                                {},
                                payload
                            ]
                        }
                    ]
                }
            }
        }
    ],
    upsert=True
)

mongo playground

因此，我们在更新中使用$替换新的根是两个对象的合并。
第一个对象是$$ root参数，因此，如果文档不存在，则基本上是一个空对象，如果它确实存在，则是原始对象，这是第一个对象的想法是保留对象上可能存在的_id字段（以及任何其他必需的PER现有字段），这就是为什么我们不仅在$ $替换中使用新对象因为它将产生一个新的_id。

第二个对象是基于使用$ cond的条件，如果现有ts更大，那么我们使用一个空对象，该对象不会更改原始$$ root，否则我们只使用新的有效载荷，该有效载荷将覆盖任何现有字段。如前所述，如果新的有效负载缺少字段，则如果可能需要进行一些较小的更改，则不会被覆盖，但是基于您的代码示例，该潜在的边缘案例没有处理，因此我认为它不存在。