WTForms:我似乎无法动态地为QuerySelectField提供默认值

时间:2012-04-06 17:17:52

标签: python sqlalchemy flask wtforms

我有一个看起来像这样的表格:

class AddProductForm(Form):
    title = TextField('Title')
    type = QuerySelectField('Type',
        query_factory=lambda: ProductType.query.order_by(ProductType.sequence).all())

    def __init__(self, formdata=None, obj=None, prefix='', **kwargs):
        try:
            product_type_id = ProductType.query.filter_by(name=obj['product_type']).one().product_type_id
            kwargs.setdefault('type', product_type_id)
        except NoResultFound:
            pass

        Form.__init__(self, formdata, obj, prefix, **kwargs)

正如您所看到的,我正在尝试将其设置为在加载表单时为product_type提供合理的默认值。但是,虽然这种代码用于设置标题作为示例,但它不适用于QuerySelectField的“类型”。有没有人有任何想法我怎么解决这个问题?

假设这是不可能的,有人知道如何动态地向表单添加表单元素吗?

2 个答案:

答案 0 :(得分:7)

要在WTForms的QuerySelectField中定义默认值,您可以直接在字段定义中执行此操作:

my_field = QuerySelectField(
    label=u"My Label",
    query_factory=lambda: Model.query(...),
    get_pk=lambda item: item.id,
    get_label=lambda item: item.name,
    allow_blank=True,
    default=lambda: Model.query.filter(...).one())

然而,它比它看起来简单一点。考虑到这一点:

  • 默认值必须是可调用
  • 此callable必须返回模型的对象实例(不是它的pk,而是整个对象)
  • 只有在没有 obj formdata 用于初始化表单实例时才会使用它(在这两种情况下,都会忽略默认值)

在您的情况下,由于您使用obj的属性初始化默认值,而您在初始化时没有这个属性,因此您可以使用工厂:

def add_product_form_factory(default_type_name):
    class AddProductForm(Form):
        title = TextField('Title')
        type = QuerySelectField('Type',
            query_factory=lambda: ProductType.query.order_by(ProductType.sequence).all(),
            default=ProductType.query.filter_by(name=default_type_name).one()
        )
    return AddProductForm

这里有一个根据参数组成表单的函数(在这种情况下,您不需要默认值是可调用的):

AddProductForm = add_product_form_factory(default_type_name='foo')
form = AddProductForm()

PS。表单工厂的好处是您可以在运行时动态处理表单,创建字​​段或设置选项。

答案 1 :(得分:2)

使用如下参数:

QuerySelectField(allow_blank=True, blank_text=u'-- please choose --', ...)

可以添加默认选择选项