天天看點

Python程式設計:Flask表單擴充Flask-WTF代碼示例

表單擴充  Flask

-WTF

文檔: 

https://flask-wtf.readthedocs.io/en/stable/ 安裝

pip install Flask-WTF      

可以進行csrf驗證

代碼示例

1、定義模型類 視圖使用

from flask import (
    Flask,
    render_template,
    session,
    url_for,
    redirect
)

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

app = Flask(__name__)

app.secret_key = "1231safdasdf"


class RegisterForm(FlaskForm):
    #                         标簽           驗證器
    user_name = StringField(label="使用者名", validators=[DataRequired("使用者名不能為空")])
    password = PasswordField(label="密碼", validators=[DataRequired("密碼不能為空")])
    re_password = PasswordField(label="密碼", validators=[DataRequired("密碼不能為空"),
                                                        EqualTo("password", "兩次密碼不一緻")])
    submit = SubmitField(label="送出")


@app.route("/register", methods=["GET", "POST"])
def register():
    # 如果是POST送出,前端資料會放到form對象中
    form = RegisterForm()

    # 驗證資料,驗證合格後再進行操作
    if form.validate_on_submit():
        username = form.user_name.data
        pwd = form.password.data
        re_pwd = form.re_password.data
        session["user_name"] = username
        return redirect(url_for("index"))

    return render_template("register.html", form=form)


@app.route("/")
def index():
    username = session.get("user_name")
    return "注冊成功:username: {}".format(username)


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

2、模闆使用

<form action="" method="POST">
    {{ form.csrf_token }}

    {{ form.user_name.label }}
    <p>{{ form.user_name }}</p>
    {% for msg in form.user_name.errors %}
    <p>{{ msg }}</p>
    {% endfor %}

    {{ form.password.label }}
    <p>{{ form.password }}</p>
    {% for msg in form.password.errors %}
    <p>{{ msg }}</p>
    {% endfor %}

    {{ form.re_password.label }}
    <p>{{ form.re_password }}</p>
    {% for msg in form.re_password.errors %}
    <p>{{ msg }}</p>
    {% endfor %}

    {{ form.submit }}
</form>      

渲染效果

Python程式設計:Flask表單擴充Flask-WTF代碼示例