Fix up activate/deactivate user commands

This commit is contained in:
Matt Wright
2012-03-09 17:13:03 -05:00
parent e239216d92
commit 7376dd353d
6 changed files with 84 additions and 66 deletions
+5 -15
View File
@@ -25,6 +25,10 @@ class MongoEngineUserDatastore(UserDatastore):
modified_at = db.DateTimeField()
return User, Role
def _save_model(self, model):
model.save()
return model
def _do_with_id(self, id):
try: return security.User.objects.get(id=id)
@@ -36,18 +40,4 @@ class MongoEngineUserDatastore(UserDatastore):
def _do_find_role(self, role):
return security.Role.objects(name=role).first()
def create_role(self, **kwargs):
role = security.Role(**self._prepare_create_role_args(kwargs))
role.save()
return role
def create_user(self, **kwargs):
user = security.User(**self._prepare_create_user_args(kwargs))
user.save()
return user
def add_role(self, user, role):
user = self._do_add_role(user, role)
user.save()
return user