notifications-api/app/dao/services_dao.py

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):
    try:
        from app.dao.permissions_dao import permission_dao
        permission_dao.remove_user_service_permissions(user, service)
        service.users.remove(user)
        db.session.add(service)
    except Exception as e:
        # Proper clean up
        db.session.rollback()
        raise e
    else:
        db.session.commit()