将 FK QuerySelectField 添加到由 model_form() 生成的 WTForm答案

【问题标题】：Adding FK QuerySelectField to WTForm generated by model_form()将 FK QuerySelectField 添加到由 model_form() 生成的 WTForm
【发布时间】：2013-06-25 20:55:39
【问题描述】：

我有一个 Flask 项目，并且正在制作与 SQLAlchemy 模型密切相关的表单。在我的 MySQL 数据库中有一个 Houses 表和一个 Garages 表。我想使用wtforms.ext.sqlalchemy.orm.model_form() 在控制器代码中动态创建我的“车库”表单，但是（这里有一个问题）将外键的选择字段添加到“房子”表中。我认为QuerySelectField() 是要走的路。

规则：所有新车库都必须有父房屋。

我正在使用 Flask-SQLAlchemy 扩展 (flask.ext.sqlalchemy.SQLAlchemy) 和 Flask-WTForms 扩展 (flask.ext.wtf.Form)，因此代码看起来与 stackoverflow 上其他地方以及 Flask、SQLAlchemy 的相应文档中的示例有点不同和 WTForms。

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
# let's use our models to make forms!
from flask.ext.wtf import Form
from wtforms.ext.sqlalchemy.orm import model_form, validators
from wtforms.ext.sqlalchemy.fields import QuerySelectField

app = Flask(__name__)
#setup DB connection with the flask-sqlalchemy plugin
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://user:pass@localhost/mydb'
db = SQLAlchemy(app)

这是我的模型：

class House(db.Model):
    __tablename__ = 'House'
    Id = db.Column(db.Integer, primary_key=True)
    Name = db.Column(db.String(32))
    Description = db.Column(db.String(128))

    @staticmethod
    def get(houseId):
        return db.session.query(House).filter_by(Id=houseId).one()

def getAllHouses():
    return House.query

这是我为页面添加车库的路由器：

@app.route("/garage/add", methods=["GET", "POST"]) #add new garage to DB
def addGarage():
    MyForm = model_form(Garage, base_class=Form,exclude_fk=False)
    garage = Garage()
    form = MyForm(request.form, garage, csrf_enabled=False)

这是我不确定的线路：

form.ParentHouse_FK = QuerySelectField(u'Assign To', query_factory=getAllHouses, get_label="Name")


    if request.method == "GET":
        return render_template('addGarage.html', form=form)
    elif form.validate_on_submit():
        form.populate_obj(garage)
        db.session.add(garage)
        db.session.commit()
    else:
        return render_template("errors.html", form=form)
    return redirect("/garage")

我的模板如下所示：

<form method="POST" action="/garage/add">
    <legend>Add Garage</legend>
    <div>{{ form.ParentHouse_FK.label }}{{ form.ParentHouse_FK }}</div>
    <div>{{ form.Name.label }}{{ form.Name(placeholder="Name") }}</div>
    <div>{{ form.Description.label }}{{ form.Description(placeholder="Description") }}</div>

    <button type="submit" class="btn">Submit</button>
</form>

注意第一个<div> 的最后一部分有form.ParentHouse_FK 没有()，否则我会收到AttributeError: 'UnboundField' object has no attribute '__call__' 错误。实际上，我仍然收到错误消息：UnboundField(QuerySelectField, (u'Assign To',), {'get_label': 'Name', 'query_factory': <function getAllHouses at 0x29eaf50>})

我的目标是在 form 中添加一个字段，以表示当前 House 行的所有可用外键可能性（然后在 Garage 表中填充 Garage.ParentHouse_FK 以获取新的 Garage 条目）。我知道我可以忽略（否则很棒）model_form 快捷方式并直接定义我的所有表单，但是这个项目的模型可能会随着时间的推移而发生变化，只需更新模型而无需更新表单代码会很棒也。理想情况下，我还会在 Jinja2 模板中有一个 for 循环来显示所有字段。

任何线索如何正确使用QuerySelectField() 和model_form() 来获得我想要的东西？

谢谢！

【问题讨论】：

标签： sqlalchemy flask-sqlalchemy wtforms flask-wtforms

【解决方案1】：

问题不在于QuerySelectField，而在于您正在尝试将字段添加到已经实例化的表单中。这需要一些特别的照顾。这是实例化表单时字段所经历的内容，您可以通过这种方式添加自己的字段。

unbound_field = QuerySelectField(...)
form._unbound_fields.append(("internal_field", unbound_field))
bound_field = unbound_field.bind(form, "field_name", prefix=form._prefix, translations=form._get_translations())
form._fields["field_name"] = bound_field
form.field_name = bound_field

但是，有一种更简单的方法。在实例化之前将字段添加到表单类中，这将自动发生。

Form = model_form(...)
unbound_field = QuerySelectField(...)
Form.field_name = unbound_field
form = Form(...)

【讨论】：