Flask WTF（表单处理）扩展详解

Flask WTF是Flask框架的一个扩展，它提供了许多方便的功能来处理表单数据。

本文将详细介绍Flask WTF的使用方法。

安装

首先，需要使用pip安装Flask WTF扩展

pip install Flask-WTF

导入

在Flask应用中，需要导入Flask-WTF模块：

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, InputRequired, Length

在这个例子中，导入了FlaskForm类，用来定义表单，并且导入了多个WTForms字段和验证器，例如StringField、PasswordField、SubmitField和DataRequired等等。

创建表单

创建表单非常简单，只需要继承FlaskForm类并定义表单字段即可。下面是一个登录表单的示例：

class LoginForm(FlaskForm):
    username = StringField('用户名', validators=[DataRequired(), Length(min=2, max=20)])
    password = PasswordField('密码', validators=[DataRequired()])
    remember = BooleanField('记住密码')
    submit = SubmitField('登录')

在这个表单中，定义了4个字段：一个文本输入框、一个密码输入框、一个复选框和一个提交按钮。

渲染表单

渲染表单是通过渲染一个表单模板来完成的。在这个模板中，将字段通过form.字段名的方式来渲染。下面是一个简单的表单模板：

<form method="post">
  {{ form.hidden_tag() }}
  <div class="form-group">
    {{ form.username.label }}
    {{ form.username(size=20, class="form-control", autofocus=True) }}
  </div>
  <div class="form-group">
    {{ form.password.label }}
    {{ form.password(size=20, class="form-control") }}
  </div>
  <div class="form-group form-check">
    {{ form.remember(class="form-check-input") }}
    {{ form.remember.label(class="form-check-label") }}
  </div>
  {{ form.submit(class="btn btn-primary") }}
</form>

这个模板使用Bootstrap来渲染表单样式。

处理表单

在Flask中处理表单的推荐方式是使用视图函数，这个函数需要定义一个路由和请求方法。下面是一个登录视图函数的示例：

@app.route('/', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if request.method == 'POST' and form.validate():
        # 验证通过
        username = form.username.data
        password = form.password.data
        remember = form.remember.data
        # 处理登录逻辑
        return redirect(url_for('home'))
    # GET请求或者验证失败
    return render_template('login.html', form=form)

在这个视图函数中，首先创建LoginForm实例。然后判断请求方法是否为POST，并且表单数据是否验证通过。如果通过验证，则可以访问表单字段的数据，并且处理登录逻辑。如果请求方法为GET或验证失败，则返回渲染后的表单模板以显示表单。

添加CSRF保护

CSRF攻击是一种常见的Web攻击，可以通过伪造用户请求来攻击Web应用。为了防止CSRF攻击，需要在表单中添加CSRF保护。Flask WTF提供了内置的CSRF保护，只需要在应用中配置SECRET_KEY即可。

app = Flask(__name__)
app.config['SECRET_KEY'] = 'secret'

然后在表单模板中添加CSRF保护：

<form method="post">
  {{ form.hidden_tag() }}
  ...
</form>

总结

本文介绍了Flask WTF的基本使用，包括创建表单、渲染表单、处理表单和添加CSRF保护。Flask WTF提供了许多方便的功能来处理表单数据，可以大大简化表单处理的过程。