WTForms:将另一个容器传递给验证器

时间:2018-06-19 17:55:55

标签: python wtforms

已经看过this的答案,我很难让我的自定义WTForm验证器接受另一个参数。

我的代码是如何结构的是,我有一个form_create()方法,该方法通过以下方式调用:

    form = other_file.form_create(
        form='page_settings',
        data=data,
        all_applications=all_applications,
    )

all_applications是我需要为我的验证程序之一建立索引的字典。 form_create如下所示:

def get_form(form=None, *args, **kwargs):
    class Form(wtf.Form):
        if form == 'page_settings':
            my_host_name = HostName('Hostname', validators=[wtf.validators.DataRequired(), validate_hostname, validate_config])

我希望validate_config使用该字段和all_applications中的数据,但即使将其放在HostName的 init 方法中,例如:

class HostName(wtf.StringField):

    def __init__(self, *args, **kwargs):
        kwargs.setdefault('validators', [
            wtf.validators.DataRequired(),
        ])
        all_applications = kwargs.setdefault('all_applications')
        super(HostNameField, self).__init__(*args, **kwargs)

但是我不断收到类似“表单”对象没有属性“ all_applications”的错误

有问题的自定义验证器如下:

def validate_config(form, field, all_applications):
    if (some logic here is true):
        raise wtf.ValidationError(
            'some error string here'
        )

关于如何让自定义验证器接受另一个all_applications之类的容器的想法吗?

1 个答案:

答案 0 :(得分:1)

您可以在初始化表单时将all_applications设置为字段的属性,然后通过验证器中的字段进行访问。

def validate_config(form, field):
    if field.data and field.all_applications is not None:
        if field.data not in field.all_applications:
            raise ValidationError('Invalid config')


class F(Form):

    foo = StringField(validators=[validate_config])

    def __init__(self, *args, **kwargs):
        all_applications = kwargs.pop('all_applications', None)
        super().__init__(*args, **kwargs)
        # Make all_applications an attribute of field foo
        self.foo.all_applications = all_applications


>>> from webob.multidict import MultiDict

>>> f = F(formdata=MultiDict(foo='a'), all_applications={'a': 1})
>>> f.validate()
True

>>> f = F(formdata=MultiDict(foo='b'), all_applications={'a': 1})
>>> f.validate()
False
>>> f.errors
{'foo': ['Invalid config']}