mirror of
https://github.com/wassname/flask-security.git
synced 2026-09-09 11:22:35 +08:00
Fix up activate/deactivate user commands
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user