2016-02-09 12:01:17 +00:00
|
|
|
from app import db
|
|
|
|
|
from app.models import Notification
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def save_notification(notification, update_dict={}):
|
|
|
|
|
if update_dict:
|
|
|
|
|
update_dict.pop('id', None)
|
|
|
|
|
update_dict.pop('job', None)
|
2016-02-09 12:48:27 +00:00
|
|
|
update_dict.pop('service', None)
|
|
|
|
|
update_dict.pop('template', None)
|
2016-02-09 12:01:17 +00:00
|
|
|
Notification.query.filter_by(id=notification.id).update(update_dict)
|
|
|
|
|
else:
|
|
|
|
|
db.session.add(notification)
|
|
|
|
|
db.session.commit()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_notification(job_id, notification_id):
|
|
|
|
|
return Notification.query.filter_by(job_id=job_id, id=notification_id).one()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_notifications_by_job(job_id):
|
|
|
|
|
return Notification.query.filter_by(job_id=job_id).all()
|