wassname/flask-security
1
0
Fork 0
mirror of https://github.com/wassname/flask-security.git synced 2026-07-14 01:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Be sure to save user model when using add_role_to_user method for MongoEngineUserDatastore. Fixes #170

Browse Source
This commit is contained in:
Matt Wright
2013-10-16 10:55:56 -04:00
parent cfd0efb91d
commit b1ac2598c0
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flask_security/datastore.py
+6
View File
@@ -234,6 +234,12 @@ class MongoEngineUserDatastore(MongoEngineDatastore, UserDatastore):
def find_role(self, role):
return self.role_model.objects(name=role).first()
def add_role_to_user(self, user, role):
rv = super(MongoEngineUserDatastore, self).add_role_to_user(user, role)
if rv:
self.put(user)
return rv
class PeeweeUserDatastore(PeeweeDatastore, UserDatastore):
"""A PeeweeD datastore implementation for Flask-Security that assumes