我有一个 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()
获得我所追求的?
谢谢!