使用相同的表单创建和更新但不同的验证逻辑

Question

我有一个 FlaskForm 类，我在其中验证字段值是否已在系统中，并在需要时引发 ValidationError。

def validate_username(self, username_field):
    if User.query.filter_by(username=username_field.data).first():
        raise ValidationError("This username is already taken.")

对于用户，我实际上不得不使用不同的验证来创建一个单独的 Form 类来更新模型。

def validate_username(self, username_field):
    if current_user.username != username_field.data:
        if User.query.filter_by(username=username_field.data).first():
            raise ValidationError("Cannot change username to this as it already is taken.")

但是现在我需要使用另一个我无法从 Form 类访问的模型，因为它与用户无关（它取决于正在编辑的记录）。我不确定如何进行此验证，如果可能的话，我不希望这些模型有两个 FlaskForm 类。

有什么方法可以将某些验证方法仅应用于某个操作（创建与更新），还是我必须为每个具有唯一字段的模型编写多个 FlaskForm 类？

我认为这种设计模式必须经常出现，但找不到处理它的好例子。

Answer 1

每个模型可以使用一个 FlaskForm。这是 User 模型的表单类：

class UserForm(FlaskForm):
    """
    Form to add or edit a user
    """

    username = StringField('username', validators=[DataRequired()])
    password = StringField('password', validators=[DataRequired()])
    ...

然后在你的views.py中你这样做：

@app.route('/user/add', methods=['GET', 'POST'])
@login_required
def add_user():
    """
    Add a user to the database
    """

    add_user = True

    form = UserForm()
    if form.validate_on_submit():
        user = User(...)
        ...
        # store the user in the db

    return render_template('/.../user.html', action="Add",
                   add_user=add_user, form=form,
                   title="Add User")


@app.route('/user/edit/<int:id>', methods=['GET', 'POST'])
@login_required
def edit_user(id):
    """
    Edit a user
    """

    add_user = False

    user = User.query.get_or_404(id)
    form = UserForm(obj=user)

    if form.validate_on_submit():
        user.username = form.username.data
        ...
        # Here you can check whether new name of the user corresponds
        # already existing user
        ...


return render_template('/.../user.html', action="Edit",
                       add_user=add_user, form=form,
                       user=user, title="Edit User")

这就是你的 user.html 模板的样子：

<div class="...">
    {% if add_user %}
        <h1>Add User</h1>
    {% else %}
        <h1>Edit User</h1>
    {% endif %}
    <br/>
    <form action="" method="post" class="form">
      {{ form.hidden_tag() }}
      {{ wtf.form_errors(form, hiddens="only") }}
      {{ wtf.form_field(form.username) }}
      {{ wtf.form_field(form.password) }}
      ...

      <div class="form-group">
        <input type="submit" value="Submit" class="btn btn-default">
        <a href="{{ url_for('app.list_users') }}" class="btn btn-default">
          Back 
        </a>
      </div>
    </form>
</div>

您使用布尔值add_user 来区分添加或编辑现有用户。可能不需要为此编写单独的验证器。

希望这会有所帮助。