Break class method for ccpo perms into two methods instead of switching on a kwarg to determine if perms are given or removed
This commit is contained in:
@@ -88,13 +88,15 @@ class Users(object):
|
||||
return user
|
||||
|
||||
@classmethod
|
||||
def update_ccpo_permissions(cls, user, add_perms=False):
|
||||
if add_perms:
|
||||
permission_sets = PermissionSets.get_all()
|
||||
else:
|
||||
permission_sets = []
|
||||
def give_ccpo_perms(cls, user):
|
||||
user.permission_sets = PermissionSets.get_all()
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
||||
user.permission_sets = permission_sets
|
||||
@classmethod
|
||||
def revoke_ccpo_perms(cls, user):
|
||||
user.permission_sets = []
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
@@ -53,6 +53,6 @@ def submit_add_new_ccpo_user():
|
||||
@user_can(Permissions.CREATE_CCPO_USER, message="create ccpo user")
|
||||
def confirm_new_ccpo_user():
|
||||
user = Users.get_by_dod_id(request.form["dod_id"])
|
||||
Users.update_ccpo_permissions(user, add_perms=True)
|
||||
Users.give_ccpo_perms(user)
|
||||
flash("ccpo_user_added", user_name=user.full_name)
|
||||
return redirect(url_for("ccpo.ccpo_users"))
|
||||
|
Reference in New Issue
Block a user