mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from sqlalchemy import exc
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import transactional
|
|
from app.models import Service, ServicePermission, SERVICE_PERMISSION_TYPES
|
|
|
|
|
|
def dao_fetch_service_permissions(service_id):
|
|
return ServicePermission.query.filter(
|
|
ServicePermission.service_id == service_id).all()
|
|
|
|
|
|
def make_service_permissions_list(service_id, permissions):
|
|
arr = []
|
|
for permission in permissions:
|
|
if permission not in SERVICE_PERMISSION_TYPES:
|
|
raise ValueError("'{}' not of service permission type: {}".format(permission, SERVICE_PERMISSION_TYPES))
|
|
|
|
service_permission = ServicePermission(service_id=service_id, permission=permission)
|
|
arr.append(service_permission)
|
|
|
|
return arr
|
|
|
|
|
|
@transactional
|
|
def dao_add_and_commit_service_permissions(service_id, permissions):
|
|
service_permissions = make_service_permissions_list(service_id, permissions)
|
|
|
|
try:
|
|
db.session.add_all(service_permissions)
|
|
db.session.commit()
|
|
except exc.IntegrityError as e:
|
|
if "duplicate key value violates unique constraint" in str(e.orig):
|
|
raise ValueError(e.orig)
|
|
raise
|
|
|
|
|
|
def dao_remove_service_permission(service_id, permission=None):
|
|
return ServicePermission.query.filter(
|
|
ServicePermission.service_id == service_id,
|
|
ServicePermission.permission == permission if permission else None).delete()
|