Добавление FK QuerySelectField в WTForm, сгенерированное model_form()

У меня есть проект Flask, и я создаю формы, которые тесно связаны с моделями SQLAlchemy. В моей базе данных MySQL есть таблица домов и таблица гаражей. Я хочу использовать 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>

Обратите внимание, что последняя часть этого 1-го <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.ParentHouse_FK в таблице Garage для новой записи Garage). Я знаю, что тогда я мог бы проигнорировать (иначе классный) ярлык model_form и определить все мои формы напрямую, но модели этого проекта, вероятно, со временем изменятся, и было бы здорово просто обновить модель без необходимости обновлять также код формы. В идеале у меня также был бы цикл for в шаблоне Jinja2 для отображения всех полей.

Любые подсказки, как правильно использовать QuerySelectField() вместе с model_form(), чтобы получить то, что мне нужно?

Спасибо!


person hamx0r    schedule 25.06.2013    source источник


Ответы (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(...)
person davidism    schedule 26.06.2013