mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-24 03:21:51 -05:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from app import db
|
|
from app.models import Service
|
|
from sqlalchemy import asc
|
|
|
|
|
|
def dao_fetch_all_services():
|
|
return Service.query.order_by(asc(Service.created_at)).all()
|
|
|
|
|
|
def dao_fetch_service_by_id(service_id):
|
|
return Service.query.filter_by(id=service_id).one()
|
|
|
|
|
|
def dao_fetch_all_services_by_user(user_id):
|
|
return Service.query.filter(Service.users.any(id=user_id)).order_by(asc(Service.created_at)).all()
|
|
|
|
|
|
def dao_fetch_service_by_id_and_user(service_id, user_id):
|
|
return Service.query.filter(Service.users.any(id=user_id)).filter_by(id=service_id).one()
|
|
|
|
|
|
def dao_create_service(service, user):
|
|
try:
|
|
from app.dao.permissions_dao import permission_dao
|
|
service.users.append(user)
|
|
permission_dao.add_default_service_permissions_for_user(user, service)
|
|
db.session.add(service)
|
|
except Exception as e:
|
|
# Proper clean up
|
|
db.session.rollback()
|
|
raise e
|
|
else:
|
|
db.session.commit()
|
|
|
|
|
|
def dao_update_service(service):
|
|
db.session.add(service)
|
|
db.session.commit()
|
|
|
|
|
|
def dao_add_user_to_service(service, user):
|
|
service.users.append(user)
|
|
db.session.add(service)
|
|
db.session.commit()
|
|
|
|
|
|
def dao_remove_user_from_service(service, user):
|
|
service.users.remove(user)
|
|
db.session.add(service)
|
|
db.session.commit()
|