Delete user folder permissions when user is removed from a service

This commit is contained in:
Katie Smith
2019-02-25 15:39:11 +00:00
parent 70cdcfff55
commit cd240f9212
3 changed files with 58 additions and 2 deletions

View File

@@ -12,6 +12,7 @@ from app.dao.dao_utils import (
version_class
)
from app.dao.service_sms_sender_dao import insert_service_sms_sender
from app.dao.service_user_dao import dao_get_service_user
from app.models import (
AnnualBilling,
ApiKey,
@@ -201,8 +202,13 @@ def dao_remove_user_from_service(service, user):
try:
from app.dao.permissions_dao import permission_dao
permission_dao.remove_user_service_permissions(user, service)
service_user = dao_get_service_user(user.id, service.id)
service_user.folders = []
service.users.remove(user)
db.session.add(service)
db.session.add_all([service, service_user])
except Exception as e:
db.session.rollback()
raise e