如何用SQLAlchemy实例化wtforms.ext.sqlalchemy.fields.QuerySelectMultipleField多对多的关系

时间:2013-09-15 17:34:56

标签: sqlalchemy flask flask-sqlalchemy wtforms flask-wtforms

我正在尝试使用wtforms.ext.sqlalchemy QuerySelectMultipleField显示复选框列表,但我无法在GET上的表单上显示模型数据。

这是我的models.py

user_permissions = db.Table('user_permissions',
    db.Column('permission_id', db.Integer, db.ForeignKey('permission.id')),
    db.Column('user_id', db.Integer, db.ForeignKey('user.id'))
)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), nullable=False, unique=True)
    email = db.Column(db.String(120), nullable=False, unique=True)
    password = db.Column(db.String(80), nullable=False)

    permissions = db.relationship('Permission', secondary=user_permissions, 
            backref=db.backref('permissions', lazy='dynamic'))

    def __init__(self, username, email, password):
        self.username = username
        self.email = email
        self.password = password

    def __repr__(self):
        return '<User %r>' % self.username


class Permission(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    desc = db.Column(db.String(80), nullable=False)

    def __init__(self, desc):
        self.desc = desc

    def __repr__(self):
        return '<Permission %r>' % self.desc

这是我的forms.py

class EditUserForm(Form):
    username = TextField('Name', validators=[DataRequired()])
    email = TextField('Email', validators=[DataRequired()])
    permissions = QuerySelectMultipleField(
            query_factory=Permission.query.all,
            get_label=lambda a: a.desc,
            widget=widgets.ListWidget(prefix_label=False),
            option_widget=widgets.CheckboxInput()
    )

这是我的views.py

@app.endpoint('edit_user')
@require_login()
@require_permission('edit_user')
def edit_user(user_id):
    user = User.query.filter_by(id=user_id).first()
    if not user:
        abort(404)
    data = {
        "username": user.username,
        "email": user.email,
        "permissions": user.permissions
    }
    form = EditUserForm(**data)

问题在于,WTForm不会显示来自user.permissions的选定值,user.permissions是一个Permission模型列表。它显示一个空复选框列表。

(Pdb) print user.permissions
[<Permission u'admin'>, <Permission u'create_user'>, <Permission u'edit_user'>]

我相当肯定问题在于“权限”数据值的结构如何,但我已经尝试了我能想到的所有可能性。任何帮助将不胜感激。

我尝试过的一些事情......

"permissions": [1, 2]
"permissions": ['1', '2']
"permissions": ['permissions-1', 'permissions-2']
"permissions": [('permissions', 1), ('permissions', 2)]
"permissions": [('permissions', '1'), ('permissions', '2')]

1 个答案:

答案 0 :(得分:3)

有几个问题。

1)您执行query_factory的方式可能会导致问题,因为它可能与单个会话/查询工厂相关联。这意味着您要比较的对象与查询工厂中的对象不在同一会话中。

更安全的方法是:

    query_factory=lambda: Permission.query.all()

2)我假设您正在使用Flask-WTF,因为您没有传递formdata部分,但如果您不是(您使用的是wtforms.Form),请执行以下操作:

form = EditUserForm(request.form)

3)您将一次性迭代器作为集合传递:

    "permissions": user.permissions

如果你真的需要通过dict自定义传入的数据,你应该这样做:

    "permissions": list(user.permissions)

但是,由于你的dict是多么基本,所以最好简单地将数据作为对象传递(这应该避免一次性使用迭代器问题)

form = EditUserForm(obj=user)

将所有这些组合在一起,您的视图应如下所示:

def edit_user(user_id):
    user = User.query.filter_by(id=user_id).first()
    if not user:
        abort(404)

    form = EditUserForm(request.form, user)
    [...]