使用Flask,SQLAlchemy和WTForms更新多对多关系?

时间:2014-08-04 11:58:25

标签: python flask sqlalchemy flask-wtforms

我想使用代表这些模型的Flask,SQLAlchemy和WTForms为多对多关系创建一个表单:

personaddress = db.Table('personaddress',
    db.Column('person', db.Integer, db.ForeignKey('person.id')),
    db.Column('address', db.Integer, db.ForeignKey('address.id'))
)

class Person(db.Model):
    __tablename__ = "person"
    id = db.Column(Integer, primary_key=True)
    name = db.Column(String, nullable=False)
    addresses = db.relationship('Address', secondary=personaddress, backref=db.backref('person', lazy='dynamic'))

class Address(db.Model):
    __tablename__ = "address"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String, nullable=False)

要求

现在我想创建一个包含表单的单个页面来实现以下目的:

  • 添加/编辑/删除某人
  • 添加/编辑/删除地址
  • 添加/编辑/删除人与地址之间的关系

重要要求:使用QuerySelectField,我可以为一个人选择现有地址。但我想以相同的形式添加新地址。

我使用model_form作为联结表的主要模型和子表单使用FormField,但我无法弄清楚如何更新所有内容,包括外键关系。页面应该有一个提交按钮,用于显示所有表单和子表单。

问题

  1. 上述要求通常如何在Flask中实施?
  2. 这是多对多场景,Django可以通过其管理界面更轻松地处理这些场景吗?

1 个答案:

答案 0 :(得分:4)

我之前也遇到过类似的事情。我试图通过使用model_form来解决它,但它并没有完全解决动态添加新条目的问题,而且在处理关系时我很难使用它。

在WTForms中使用QuerySelectField只会帮助您填充例如。 <选择>具有对应于现有地址的id,值对。但它仍然会在模板中呈现为常规的html表单。

通过使用某种多选项,可以在前端中dynamically add个新选项,您可以以相同的形式发送其他地址。如果端点不存在,则端点将负责创建新的地址。

WTForm表格将是:

from app import db

class PersonAddressForm(Form):
    id = HiddenField('id')
    name = StringField('Name')
    addresses = QuerySelectField('Addresses', 
            query_factory=lambda: db.session.query(Address), 
            get_pk=lambda a: a.id, get_label=lambda a: a.name)

    # Custom validate
    def validate(self):
        # ... custom validation
        return True

路线就像:

# ... this will be used to create and update a user
@route('create/<userid>', methods=["GET"])
def get_user_form(userid):
    # ... Get the Person
    user = Person()
    if userid:
        # ... if userid supplied, use existing Person object
        user = Person.query.get(userid)

    # ... Populate the form
    person_form = PersonAddressForm(obj=user)

    # ... return form
    return render_template('somepage.html', form=person_form)

@route('create/<userid>', methods=["POST"])
def post_person_form(userid):
    person_form = PersonAddressForm(request.form)

    if person_form.validate():
        # ... Get db object
        person = db.session.query(Person).get(form.id)

        # ... Add changes to the object from the form
        person_form.populate_obj(obj=person_address)

        # ... Get addresses
        addresses = form.addresses.raw_data

        # ... loop over and add to person
        for address in addresses:
            # Add or create an address
            actual_address = db.session.query(Address).get(address.id)

            # ... check if address is existing
            if not actual_address:
                # ... if address not existing, create new one
                actual_address = Address(address.name)
                db.session.add(actual_address)

            # ... Append new or created address to person
            person.addresses.append(actual_address)

        # ... save changes to the db
        db.session.commit()

        # ... Update/Create complete
        return redirect(url_for('get_users'))

    else:
        # ... form not valid, notify user
        # ...

这将处理编辑/创建用户并创建地址。以及创造之间的关系。要使它也支持删除地址,请更改

person.addresses.append(actual_address)

person.addresses = list_of_actual_addresses

并在人物模型中进行更改(cascade =&#39; delete-orphan&#39;)

addresses = db.relationship('Address', secondary=personaddress, cascade='delete-orphan' backref=db.backref('person', lazy='dynamic'))

这将使表单每次更新整个地址关系,级联将删除孤立的地址。因此,每次提交表单时,都会更新一个人的整个地址列表。

在模板中处理WTForms时,强烈建议您使用宏,如果您还没有。您必须在某种程度上重写它,但请检查this

希望这有帮助