动态选择WTForms Flask SelectField

时间:2017-10-25 00:02:05

标签: python flask flask-wtforms

我尝试使用FlaskForms将userID变量传递给WTForms。首先,我将展示运行良好的代码,然后展示我需要修改的代码(我不知道如何处理)。我添加了与某个群组相关联的新名称。

FlaskForm模型:

class AddName(FlaskForm):
    name =StringField('Device name', validators=[InputRequired(),Length(min=4, max=30)])
    groupID = SelectField('Payload Type', choices=[(1,"Group1"),(2,"Group2")], validators=[InputRequired])

查看型号:

@app.route('/dashboard/addname', methods=['GET', 'POST'])
def addname():
    form=AddName()
    if form.validate_on_submit():
        name=Name(form.name.data,form.groupID.data)
        db.session.add(name)
        db.session.commit()
        return "New name added"

模板:

              <form method="POST" action="/dashboard/addname">
                  <h2>Add name</h2>
                  {{ form.hidden_tag() }}
                  {{ wtf.form_field(form.name) }}
                  {{ wtf.form_field(form.groupID) }}
                  <button type="submit">Add name</button>
              </form>

我在下拉列表中看到了正确的列表,并在提交时给出了正确的数字。

任务:我需要根据 current_user.userID 传递不同的列表。 我使用SQLAlchemy形成列表,通过从DB中的表进行查询,所以我的 Flask视图是:

@app.route('/dashboard/addname', methods=['GET', 'POST'])
def addname():
    available_groups=db.session.query(Groups).filter(Groups.userID == currend_user.userID).all()
    #Now forming the list of tuples, so it's ok for SelectField
    groups_list=[(i.groupID, i.groupName) for i in available_groups]
    form=AddName()
    if form.validate_on_submit():
        name=Name(form.name.data,form.groupID.data)
        db.session.add(name)
        db.session.commit()
        return "New name added"
  1. 如何将 groups_list 传递给表单?我试图在FlaskForm模型中实现形成过程,但它没有看到current_user对象
  2. 我是否需要将groupID转换为 string ,然后在需要将其传递给SelectField(如元组)时再返回 int

1 个答案:

答案 0 :(得分:5)

这里的主要思想是在实例化后将选择列表分配给字段。为此,您需要使用参数coerce=int。 SelectField的coerce关键字arg表示我们使用int()来强制表单数据。默认的强制是unicode()。

更正FormModel:

class AddName(FlaskForm):
    name =StringField('Device name', validators=[InputRequired(),Length(min=4, max=30)])
    groupID = SelectField('Payload Type', coerce=int, validators=[InputRequired])

正确观点

@app.route('/dashboard/addname', methods=['GET', 'POST'])
def addname():
    available_groups=db.session.query(Groups).filter(Groups.userID == currend_user.userID).all()
    #Now forming the list of tuples for SelectField
    groups_list=[(i.groupID, i.groupName) for i in available_groups]
    form=AddName()
    #passing group_list to the form
    form.groupID.choices = groups_list
    if form.validate_on_submit():
        name=Name(form.name.data,form.groupID.data)
        db.session.add(name)
        db.session.commit()
        return "New name added"