I've a small problem on my App build with Flask Framework.
I'm trying to create a simple User + Permissions module. To archive it, I've a many-to-many relation between Users and Permissions table.
Here is my model, form and route
Model
user_perm = db.Table('user_perm',
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('perm_id', db.Integer, db.ForeignKey('permissions.id'))
)
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(32), unique=True, nullable=False)
pwdhash = db.Column(db.String(16), nullable=False)
email = db.Column(db.String(128))
permissions = db.relationship('Permission', secondary=user_perm, backref=db.backref('users', lazy='dynamic'))
def __init__(self, username, pwdhash, email, perms):
self.username = unicode(username)
self.pwdhash = pwdhash
self.email = email
self.permissions = perms
def __repr__(self):
return '<User {0}>'.format(self.username)
class Permission(db.Model):
__tablename__ = 'permissions'
id = db.Column(db.Integer)
perm = db.Column(db.String(50), primary_key=True, unique=True, nullable=False)
def __init__(self, perm):
self.perm = unicode(perm)
def __repr__(self):
return '<Permission {0}>'.format(self.perm)
Form
class AddUser(Form):
username = TextField(u'Username', required)
pwdhash = TextField(u'Password', required)
email = TextField(u'E-email', email_validators)
permissions = SelectMultipleField(u'Permissions', required, coerce=int)
class EditUser(AddUser):
pass
Routes
@app.route('/admin/user/add', methods=['GET', 'POST'])
@login_required
def admin_user_add():
form = AddUser(request.form)
form.permissions.choices = [(p.id, p.perm) for p in Permission.query.order_by('perm')]
if request.method == 'POST' and form.validate():
user = User(
form.username.data,
form.pwdhash.data,
form.email.data,
Permission.query.filter(Permission.id.in_(form.permissions.data)).all()
)
db.session.add(user)
db.session.commit()
flash('Successfully added user', category='success')
return redirect(url_for('users'))
return render_template('admin_user_add.html', form=form)
@app.route('/admin/user/edit/<int:user_id>', methods=['GET', 'POST'])
@login_required
def admin_user_edit(user_id):
user = User.query.filter_by(id=user_id).first_or_404()
form = EditUser(request.form, obj=user)
form.permissions.choices = [(p.id, p.perm) for p in Permission.query.order_by('perm')]
form.permissions.data = [p.id for p in user.permissions]
if request.method == 'POST' and form.validate():
form.populate_obj(user)
user.username = form.username.data
user.pwdhash = form.pwdhash.data
user.email = form.email.data
user.permissions = Permission.query.filter(Permission.id.in_(form.permissions.data)).all()
db.session.merge(user)
db.session.commit()
flash('Successfully updated user', category='success')
return redirect(url_for('users'))
return render_template('admin_user_add.html', form=form, edit=True)
The adding actions works perfectly, the relation are been add, but when I'm trying to edit an user, I get this error.
Traceback
Traceback (most recent call last):
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1518, in __call__
return self.wsgi_app(environ, start_response)
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1506, in wsgi_app
response = self.make_response(self.handle_exception(e))
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1504, in wsgi_app
response = self.full_dispatch_request()
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1264, in full_dispatch_request
rv = self.handle_user_exception(e)
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1262, in full_dispatch_request
rv = self.dispatch_request()
File "/usr/local/lib/python2.7/dist-packages/flask/app.py", line 1248, in dispatch_request
return self.view_functions[rule.endpoint](**req.view_args)
File "/var/www/deploy.staging.inovae.ch/webapp/webapp/utilities.py", line 29, in inner
return f(*args, **kwargs)
File "/var/www/deploy.staging.inovae.ch/webapp/webapp/utilities.py", line 44, in wrapper
return f(*args, **kwargs)
File "/var/www/deploy.staging.inovae.ch/webapp/webapp/views.py", line 161, in admin_user_edit
form.populate_obj(user)
File "/usr/local/lib/python2.7/dist-packages/wtforms/form.py", line 73, in populate_obj
field.populate_obj(obj, name)
File "/usr/local/lib/python2.7/dist-packages/wtforms/fields/core.py", line 283, in populate_obj
setattr(obj, name, self.data)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/attributes.py", line 155, in __set__
instance_dict(instance), value, None)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/attributes.py", line 892, in set
lambda adapter, i: adapter.adapt_like_to_iterable(i))
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/attributes.py", line 927, in _set_iterable
collections.bulk_replace(new_values, old_collection, new_collection)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/collections.py", line 681, in bulk_replace
new_adapter.append_with_event(member)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/collections.py", line 555, in append_with_event
getattr(self._data(), '_sa_appender')(item, _sa_initiator=initiator)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/collections.py", line 945, in append
item = __set(self, item, _sa_initiator)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/collections.py", line 920, in __set
item = getattr(executor, 'fire_append_event')(item, _sa_initiator)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/collections.py", line 614, in fire_append_event
item, initiator)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/attributes.py", line 800, in fire_append_event
value = fn(state, value, initiator or self)
File "/usr/local/lib/python2.7/dist-packages/sqlalchemy/orm/unitofwork.py", line 35, in append
item_state = attributes.instance_state(item)
AttributeError: 'long' object has no attribute '_sa_instance_state'
The crux of your problem is that SQL Alchemy is expecting the permissions
relationship to be populated with sqlalchemy permission
objects, not the int
s you are passing it.
In your view:
form = EditUser(request.form, obj=user)
form.permissions.choices = [(p.id, p.perm) for p in Permission.query.order_by('perm')]
form.permissions.data = [p.id for p in user.permissions]
So in your form, form.permissions
is actually just holding the id
(int) of the permission object's, not the objects themselves. When you call form.populate_obj(user)
, when it gets to the permissions segment, its going to call this method:
setattr(obj, name, self.data)
Which translates to:
setattr(user, permissions, sefl.data)
Which translates to:
user.permissions = [1, 3, 4 ...]
But SQL Alchemy expects:
user.permissions = [sqlalchemyobject1, sqlalchemyobject2, ...]
There are at least a few ways to solve this problem. The method I use is sqlalchemy's Association Proxy. This will let you do things like pass a list of integers to your user model, and have the Model convert them to Permissions objects:
class User(db.Model):
...
permissions_relationship = ('Permission', cascade="all,delete,delete-orphan")
permissions = association_proxy('permissions_relationship', 'perm',
creator=lambda perm: Permission(perm=perm)
class Permission(db.Model):
...
perm = ...
def __init__(self,perm):
self.perm = perm
In a Nutshell, the association_proxy
call simply says: "When you get passed a list of items (Ints, Strings, whatever), convert them to Permission
objects by doing this: Permission(perm=perm)
. Using this solution in your above code, you would remove the line form.permissions.data = [p.id for p in user.permissions]
.
Finally, wtforms
also provides a QuerySelectMultiple field. I haven't used it, but imagine it would also serve as a solution to this problem.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With