wassname/flask-security
1
0
Fork 0
mirror of https://github.com/wassname/flask-security.git synced 2026-07-18 12:20:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #40

Browse Source
This commit is contained in:
Matt Wright
2012-11-26 14:41:51 -05:00
parent 9e8fcbc879
commit 7f002ac612
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flask_security/datastore.py
+5 -2
View File
@@ -56,8 +56,11 @@ class UserDatastore(object):
self.role_model = role_model
def _prepare_role_modify_args(self, user, role):
role = role.name if isinstance(role, self.role_model) else role
return self.find_user(email=user.email), self.find_role(role)
if isinstance(user, basestring):
user = self.find_user(email=user.email)
if isinstance(role, basestring):
role = self.find_role(role)
return user, role
def _prepare_create_user_args(self, **kwargs):
kwargs.setdefault('active', True)