from app import db
from app.models import Notification


def dao_create_notification(notification):
    db.session.add(notification)
    db.session.commit()


def dao_update_notification(notification):
    db.session.add(notification)
    db.session.commit()


def get_notification_for_job(service_id, job_id, notification_id):
    return Notification.query.filter_by(service_id=service_id, job_id=job_id, id=notification_id).one()


def get_notifications_for_job(service_id, job_id):
    return Notification.query.filter_by(service_id=service_id, job_id=job_id).all()


def get_notification(service_id, notification_id):
    return Notification.query.filter_by(service_id=service_id, id=notification_id).one()