Python中使用WTForms库创建并验证LoginForm()表单

发布时间：2024-01-03 22:57:26

WTForms是一个用于创建和验证表单的Python库。它提供了一组简单而强大的工具，可以轻松地生成HTML表单，并提供了验证和处理用户输入的功能。

首先，我们需要安装WTForms库。可以使用pip命令进行安装:

pip install wtforms

接下来，我们可以创建一个LoginForm()类，这个类将继承自wtforms.Form类。在这个类中，我们可以定义表单字段，并为每个字段添加验证规则。

让我们看一个例子，创建一个简单的登录表单。

from wtforms import Form, StringField, PasswordField, validators

class LoginForm(Form):
    username = StringField('Username', [validators.DataRequired()])
    password = PasswordField('Password', [validators.DataRequired()])

在上面的例子中，我们使用StringField和PasswordField定义了两个表单字段，分别用于输入用户名和密码。validators.DataRequired()验证函数表示该字段为必填项。

在表单的初始化函数中，我们可以传递参数作为字段的标签名。例如，上面的例子中，我们将"Username"作为用户名字段的标签名。

现在我们已经定义了表单，接下来就可以使用它了。下面是一个简单的例子，演示了如何使用LoginForm表单。

from flask import Flask, render_template, request
from wtforms import Form, StringField, PasswordField, validators

app = Flask(__name__)

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm(request.form)
    if request.method == 'POST' and form.validate():
        username = form.username.data
        password = form.password.data
        # 做一些验证等处理
        return "登录成功"
    return render_template('login.html', form=form)

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

在上面的例子中，我们定义了一个/login的路由，用于处理登录请求。在请求到达时，我们会创建一个LoginForm实例，并传递request.form作为参数。然后，我们检查请求的方法是否为POST，并且验证表单是否有效。如果验证通过，我们可以通过form.username.data和form.password.data获取表单字段的值，然后可以进行进一步的处理操作。

在页面的渲染中，我们将表单作为参数传递到login.html模板中。在模板中可以方便地通过form.username和form.password获取表单字段，并使用form.username.label来显示字段的标签名。

<!DOCTYPE html>
<html>
<head>
    <title>Login</title>
</head>
<body>
    <h1>Login</h1>
    <form action="/login" method="post">
        {{ form.csrf_token }}
        <div>
            {{ form.username.label }} {{ form.username }}
        </div>
        <div>
            {{ form.password.label }} {{ form.password }}
        </div>
        <div>
            <input type="submit" value="Login">
        </div>
    </form>
</body>
</html>

在上面的模板中，我们使用了form.csrf_token字段，用于防止CSRF攻击。通过{{ form.username.label }}和{{ form.username }}，我们可以将表单字段的标签和输入框渲染到表单中。

这只是WTForms库的一小部分功能。它还提供了许多其他功能，如校验数据类型、验证字符串长度、验证邮箱地址等。