Ad a reference to the model

- used if 3rd party needs to record an ID for reconciliation purposes

app/dao/notifications_dao.py

@@ -64,6 +64,7 @@ def update_job_sent_count(notification):
 
 
 def dao_update_notification(notification):
+    notification.updated_at = datetime.utcnow()
     db.session.add(notification)
     db.session.commit()
 
@@ -90,6 +91,28 @@ def update_notification_status_by_to(to, status):
     return count
 
 
+def update_notification_status_by_reference(reference, status):
+    count = db.session.query(Notification).filter_by(
+        refernce=reference
+    ).update({
+        Notification.status: status,
+        Notification.updated_at: datetime.utcnow()
+    })
+    db.session.commit()
+    return count
+
+
+def update_notification_reference_by_id(id, reference):
+    count = db.session.query(Notification).filter_by(
+        id=id
+    ).update({
+        Notification.reference: reference,
+        Notification.updated_at: datetime.utcnow()
+    })
+    db.session.commit()
+    return count
+
+
 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()