使用 wtforms 验证 - 检查 MongoDB 集合中的现有记录

Question

我正在尝试使用 Flask 和 Mongodb 的应用程序。 我使用 WTForms 和 Flask-login，当我使用这种方法注册时，我目前正在验证用户名是否重复：

def validate_username(self, username):
    user = mongo.db.users.find_one({"username": username.data})
    if user:
        raise ValidationError("The username already exists.")

我在编辑用户时添加了第二行，因为我使用的是flask-login，而我是current_user：

def validate_username(self, username):
    if username.data != current_user.username:
        user = mongo.db.users.find_one({"username": username.data})
        if user:
            raise ValidationError("The username already exists.")

这一切都很好，但我还想验证另一个表单中的另一个字段，当然，flask-login 中不存在该字段。这在我创建 typeName 时效果很好：

def validate_type_name(self, type_name):
    typeName = mongo.db.types.find_one({"type_name": type_name.data})
    if typeName:
        raise ValidationError("The type name already exists.")

但是：当我 EDIT 和 typeName 时，我如何检查重复项？ 检查我的函数第二行的神奇方法是什么？或者有什么办法？

这是我的edit_type 路线：

@app.route("/type/edit/<id>", methods=["POST", "GET"])
@login_required
def edit_type(id):
    form = EditTypeForm()
    type = mongo.db.types.find_one({"_id": ObjectId(id)})
    if current_user.is_admin:
        if form.validate_on_submit():
            new_value = {
                "$set": {
                    "type_name": form.type_name.data,
                    "description": form.description.data,
                }
            }
            mongo.db.types.update_one(type, new_value)
            flash("Type has been updated", "info")
            return redirect(url_for("type", id=type["_id"]))
        elif request.method == "GET":
            form.type_name.data = type["type_name"]
            form.description.data = type["description"]
    return render_template("edit_type.html", title="Edit Type", form=form)

请帮忙？

谢谢！

Answer 1

这个问题通过在表单中​​添加一个隐藏字段并为其分配与数据库中当前相同的值来解决。然后在验证器中有一些东西需要检查。 完整答案在这里：

Check duplication when edit an exist database field with WTForms custom validator

Answer 2

为了在只能编辑描述和保持类型名称不变的情况下进行更改，如果新类型名称已经存在则给你一个错误，你可以使用函数validate_type_name。

def validate_type_name(self, type_name):
    typeName = mongo.db.types.find_one({"type_name": type_name.data})
    if typeName:
        return False
    else:
        return True

然后制作你的 edit_type ：

@app.route("/type/edit/<id>", methods=["POST", "GET"])
@login_required
def edit_type(id):
    form = EditTypeForm()
    type = mongo.db.types.find_one({"_id": ObjectId(id)})
    if current_user.is_admin:
        if form.validate_on_submit():
            new_value = {
                "$set": {
                    "type_name": form.type_name.data,
                    "description": form.description.data,
                }
            }
            if not mongo.db.types.find_one({"type_name": form.type_name.data}):
               mongo.db.types.update_one(type, new_value)
               flash("Type has been updated", "info")
               return redirect(url_for("type", id=type["_id"]))
            else:
               <Write validation error message here>
        elif request.method == "GET":
            form.type_name.data = type["type_name"]
            form.description.data = type["description"]
    return render_template("edit_type.html", title="Edit Type", form=form)