确定在Flask视图中按下了哪个WTForms按钮

时间:2016-03-03 13:51:44

标签: python flask wtforms flask-wtforms

我有一个包含多个链接的页面,可以将用户重定向到不同的页面。我认为使用表单会更好,所以我定义了一个带有多个Form的WTForms SubmitFields。如何确定单击了哪个按钮并根据该按钮重定向?

class MainForm(Form):
    user_stats = SubmitField('User Stats')
    room_stats = SubmitField('Room Stats')

@main.route('/')
@login_required
def index():
    form = MainForm()    
    return render_template('index.html', form=form)
<form action="#" method="post">
    {{ wtf.quick_form(form) }}
</form>

1 个答案:

答案 0 :(得分:12)

您在表单中添加了两个按钮,因此请检查哪些字段的数据为True

from flask import Flask, render_template, redirect, url_for
from flask_wtf import Form
from wtforms import SubmitField

app = Flask(__name__)
app.secret_key = 'davidism'

class StatsForm(Form):
    user_stats = SubmitField()
    room_stats = SubmitField()

@app.route('/stats', methods=['GET', 'POST'])
def stats():
    form = StatsForm()

    if form.validate_on_submit():
        if form.user_stats.data:
            return redirect(url_for('user_stats'))
        elif form.room_stats.data:
            return redirect(url_for('room_stats'))

    return render_template('stats.html', form=form)

app.run(debug=True)
<form method="post">
    {{ form.hidden_tag() }}
    {{ form.user_stats }}
    {{ form.room_stats }}
</form>