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

Question

我有一个包含多个链接的页面，用于将用户重定向到不同的页面。我认为使用表单会更好，所以我定义了一个WTForms Form与多个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> 

Answer 1

您添加了两个按钮的形式，因此检查其中的字段数据是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>