Flask表单处理与验证

Flask是一个轻量级的Python框架，它通过扩展库提供了对表单处理与验证的支持。WTForms是一个流行的Flask扩展库，用于创建和验证Web表单。它提供了一种声明式的方法来定义表单结构和验证逻辑，使得表单处理更为简洁和优雅。下面，我们来深入探讨如何使用WTForms进行表单处理和数据验证。

安装WTForms

首先，你需要在激活的虚拟环境中安装WTForms：

pip install WTForms

表单类和字段定义

使用WTForms时，我们主要通过定义表单类来进行表单处理。一个WTForms表单是一个Python类，其中包含多个字段，每个字段对应一个类型（如字符串、整数、选择等）。

定义一个基本的表单

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

class LoginForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired(), Length(min=4, max=25)])
    password = PasswordField('Password', validators=[DataRequired()])
    submit = SubmitField('Log In')

在这个例子中，LoginForm 是一个继承自 FlaskForm 的类（FlaskForm 是 WTForms 和 Flask 集成的一个集成类），其中定义了 username 和 password 字段及一个 submit 按钮。字段参数validators用于指定验证器，当用户提交表单时这些验证器会自动检查用户输入是否符合要求。

数据验证和错误处理

WTForms使用验证器来确保输入数据的有效性。你可以使用WTForms内置的验证器，也可以自定义验证器。

常用验证器

DataRequired：确保字段不为空。
Length：限制字段输入的长度。
Email：验证输入是否为有效的Email格式。
EqualTo：用于比较两个字段的值是否相等。

表单处理流程

表单验证和处理通常是在视图函数中完成的：

from flask import Flask, render_template, redirect, url_for, flash
from forms import LoginForm # 假设表单定义在 forms.py
from flask_wtf.csrf import CSRFProtect

app = Flask(__name__)
app.secret_key = 'your_secret_key_here'
csrf = CSRFProtect(app)  # 保护表单免受CSRF攻击

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        username = form.username.data
        password = form.password.data
        # 在这里处理登录逻辑
        flash('Login successful!', 'success')
        return redirect(url_for('index'))
    return render_template('login.html', form=form)

if __name__ == '__main__':
    app.run(debug=True)

在这个视图函数中：

LoginForm()实例化为一个表单对象。
validate_on_submit()方法检查请求是否是POST，以及所有的验证器是否通过。如果是，它将返回True。
否则，处理GET请求，或者渲染表单时发生错误时，重呈表单并显示验证错误。

渲染表单和显示错误

在HTML模板中，可以方便地使用表单包含的字段和验证错误信息：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Login</title>
</head>
<body>
    <h1>Login</h1>
    <form method="POST" action="">
        {{ form.hidden_tag() }}
        <p>
            {{ form.username.label }}<br>
            {{ form.username(size=32) }}<br>
            {% for error in form.username.errors %}
                <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.password.label }}<br>
            {{ form.password(size=32) }}<br>
            {% for error in form.password.errors %}
                <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>{{ form.submit() }}</p>
    </form>
</body>
</html>

使用 form.hidden_tag() 来避免跨站请求伪造（CSRF）攻击，这也是为什么需要定义 app.secret_key。
遍历 form.<field>.errors 来显示字段的验证错误。

通过这些步骤，你可以用WTForms在Flask中实现高级的表单处理与验证。WTForms支持的功能还包括自定义验证器、表单内嵌套等，面对更复杂的表单需求时，也可以很容易地扩展。