python烧瓶mongodb插入数据

Question

如何在 python flask mongodb 中向特定用户插入数据

{
    "_id" : ObjectId("5e208aa1f86973bbd7db6e8a"),
    "worker_name" : "user1",
    "location" : "location1",
    "detection" : [ ]
},
{
    "_id" : ObjectId("5e208aa1f86973bbd7db6e8b"),
    "worker_name" : "user2",
    "location" : "location2",
    "detection" : [ ]
}

上面是我的用户，我想在user1 detection列表中插入一些数据，下面是我尝试过的代码

def face_detection():
    face_module = mongo.db.face_modules
    user = mongo.db.users
    stream_link = request.form['stream_link']
    location = request.form['location']
    camera = request.form['camera']
    result = {
            "location": location,
            "stream_url": stream_link,
            "worker_name": "user1",
            "date": "1/1/2020",
            "hour": "9",
            "minute": "10",
            "second": "25"
        }
    if user.find({"worker_name": result['worker_name']}).count() > 0:
        update_user = user.detection.insert(result)
        output = "user updated"
        return jsonify({'result': output})

Answer 1

由于文档已经存在，您需要更新它而不是插入，此外，由于您想将某些内容插入到数组中，它应该通过 $push，另外，您可以使用 find_one_and_update 而不是进行两次 DB 调用，这将使用此选项返回更新的文档 :: return_document = ReturnDocument.AFTER 或在未找到匹配文档的情况下返回 none。基于此，您可以返回响应。通常，您将使用insert 或insert_one 将新文档插入到colleciton。我对pymongo 有点陌生，请添加代码以检查 DB 中的错误场景，另外测试此代码，您可以随时根据任何发现更新此答案..

试试这个：

def face_detection():
face_module = mongo.db.face_modules
user = mongo.db.users
stream_link = request.form['stream_link']
location = request.form['location']
camera = request.form['camera']
result = {
        "location": location,
        "stream_url": stream_link,
        "worker_name": "user1",
        "date": "1/1/2020",
        "hour": "9",
        "minute": "10",
        "second": "25"
    }
resp = user.find_one_and_update(
    {"worker_name": result['worker_name']},
    { '$push': {'detection' : result} },
    return_document = ReturnDocument.AFTER)


if resp :
    return jsonify({'result': resp})
else :
    return jsonify({'result': 'No document found'})