Fix up activate/deactivate user commands

flask_security/datastore/__init__.py

@@ -6,6 +6,10 @@ class UserDatastore(object):
     """Abstracted user datastore. Always extend this and implement
     missing methods"""
     
+    def _save_model(self, model, **kwargs):
+        raise NotImplementedError(
+            "User datastore does not implement _save_model method")
+    
     def _do_with_id(self, id):
         raise NotImplementedError(
             "User datastore does not implement _do_with_id method")
@@ -30,6 +34,20 @@ class UserDatastore(object):
             user.roles.remove(role)
         return user
     
+    def _do_toggle_active(self, user, active=None):
+        user = self.find_user(user)
+        if active is None:
+            user.active = not user.active
+        elif active != user.active:
+            user.active = active
+        return user
+    
+    def _do_deactive_user(self, user):
+        return self._do_toggle_active(user, False)
+    
+    def _do_active_user(self, user):
+        return self._do_toggle_active(user, True)
+    
     def _prepare_role_modify_args(self, user, role):
         if isinstance(user, security.User):
             user = user.username or user.email
@@ -92,18 +110,22 @@ class UserDatastore(object):
         if role: return role
         raise security.RoleNotFoundError()
     
-    def create_role(self, **kwargs):
-        raise NotImplementedError(
-            "User datastore does not implement create_role method")
-        
-    def create_user(self, **kwargs):
-        raise NotImplementedError(
-            "User datastore does not implement create_user method")
-        
+    def create_role(self, commit=True, **kwargs):
+        role = security.Role(**self._prepare_create_role_args(kwargs))
+        return self._save_model(role)
+    
+    def create_user(self, commit=True, **kwargs):
+        user = security.User(**self._prepare_create_user_args(kwargs))
+        return self._save_model(user)
+    
     def add_role_to_user(self, user, role):
-        raise NotImplementedError(
-            "User datastore does not implement add_role_to_user method")
-        
-    def remove_role_from_user(self, user, role):
-        raise NotImplementedError(
-            "User datastore does not implement remove_role_from_user method")
+        return self._save_model(self._do_add_role(user, role))
+    
+    def remove_role_from_user(self, user, role, commit=True):
+        return self._save_model(self._do_remove_role(user, role))
+    
+    def deactivate_user(self, user):
+        return self._save_model(self._do_deactive_user(user))
+    
+    def activate_user(self, user, commit=True):
+        return self._save_model(self._do_active_user(user))

flask_security/datastore/mongoengine.py

@@ -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
+    

flask_security/datastore/sqlalchemy.py

@@ -49,9 +49,9 @@ class SQLAlchemyUserDatastore(UserDatastore):
             
         return User, Role
     
-    def _save_model(self, model, commit=True):
+    def _save_model(self, model):
         self.db.session.add(model)
-        if commit: self.db.session.commit()
+        self.db.session.commit()
         return model
     
     def _do_with_id(self, id):
@@ -63,20 +63,4 @@ class SQLAlchemyUserDatastore(UserDatastore):
     
     def _do_find_role(self, role):
         return security.Role.query.filter_by(name=role).first()
-    
-    def create_role(self, commit=True, **kwargs):
-        role = security.Role(**self._prepare_create_role_args(kwargs))
-        return self._save_model(role, commit)
-    
-    def create_user(self, commit=True, **kwargs):
-        user = security.User(**self._prepare_create_user_args(kwargs))
-        return self._save_model(user, commit)
-    
-    def add_role_to_user(self, user, role, commit=True):
-        user = self._do_add_role(user, role)
-        return self._save_model(user, commit)
-    
-    def remove_role_from_user(self, user, role, commit=True):
-        user = self._do_remove_role(user, role)
-        return self._save_model(user, commit)