客户端部分生成表单时如何进行Flask WTForms表单验证

时间:2019-05-31 12:14:44

标签: python flask flask-wtforms wtforms

我正在构建一个带有两个下拉菜单的简单用户界面。根据第一个下拉列表中选择的内容,第二个下拉列表通过jquery / AJAX填充了一些新值。问题在于,提交表单后,表单验证正确地失败了,因为返回的第二个下拉列表中的选项不在生成该下拉列表的列表中。

我通过在验证表单之前将从下拉列表2发送的所有数据注入到表单对象中来使其工作。这段代码如下所示,但感觉像是一个骇人听闻的解决方案。

form = SearchForm(request.form)

# dropdown 1
form.node_type.choices = [('default', 'Please select a node type')]
form.node_type.choices.extend([(t, t) for t in node_types()])

# dropdown 2
form.node_name.choices = [('default', 'Select a node type first')]

if request.method == 'POST':
    # the below 2 lines are the hack that make it work.
    node_name = request.form.get('node_name')
    form.node_name.choices.append((node_name, node_name))

    if form.validate():
        print('hooray')

是否有一种标准/推荐的方法可以在客户端动态添加选项,并且仍然让WTForms正确执行表单验证?

1 个答案:

答案 0 :(得分:0)

这是纯wtforms解决方案,但应与flask-wtf配合使用(flask-wtf在form.validate()中调用form.validate_on_submit())。

如果Form.validate字段的验证失败并显示“无效选择”错误,则代码将node_name扩展为调用自定义验证函数。如果验证函数返回True,则会从Form.errors中删除“不是有效选择”错误,如果True现在为空,则返回Form.errors

import werkzeug
import wtforms


class Form(wtforms.Form):

    node_type = wtforms.SelectField(choices=[(x, x) for x in 'ABCDEF'])
    node_name = wtforms.SelectField(choices=[('', '')])

    def validate(self):
        if super().validate():
            return True
        invalid_choice = 'Not a valid choice'
        if 'node_name' in self.errors and invalid_choice in self.errors['node_name']:
            is_valid_node_name = self.cross_validate_node_name()
            if is_valid_node_name:
                self.errors['node_name'].remove(invalid_choice)
                if not self.errors['node_name']:
                    del self.errors['node_name']
                return not self.errors
        return False

    def cross_validate_node_name(self):
        # Implement your validation logic here.
        return self.node_type.data == 'B' and self.node_name.data == 'spam'


if __name__ == '__main__':
    datas = [werkzeug.MultiDict([('node_type', 'A'), ('node_name', 'spam')]),
             werkzeug.MultiDict([('node_type', 'B'), ('node_name', 'spam')])]
    for data in datas:
        form = Form(formdata=data)
        if form.validate():
            print('OK')
        else:
            print(form.errors)