darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-07 06:58:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix core daos

Browse Source
This commit is contained in:
Kenneth Kehl
2024-10-11 10:13:11 -07:00
parent 958861df59
commit c44300a737
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/dao/notifications_dao.py
View File

@@ -465,9 +465,7 @@ def dao_timeout_notifications(cutoff_time, limit=100000):
stmt = ( stmt = (
update(Notification) update(Notification)
.filter(Notification.id.in_([n.id for n in notifications])) .filter(Notification.id.in_([n.id for n in notifications]))
.values( .values({"status": new_status, "updated_at": updated_at})
{"status": new_status, "updated_at": updated_at}
)
) )
db.session.execute(stmt) db.session.execute(stmt)
@@ -482,7 +480,8 @@ def dao_update_notifications_by_reference(references, update_dict):
.filter(Notification.reference.in_(references)) .filter(Notification.reference.in_(references))
.values(update_dict) .values(update_dict)
) )
updated_count = db.session.execute(stmt).scalar() or 0 result = db.session.execute(stmt)
updated_count = result.rowcount
updated_history_count = 0 updated_history_count = 0
if updated_count != len(references): if updated_count != len(references):
@@ -491,7 +490,8 @@ def dao_update_notifications_by_reference(references, update_dict):
.filter(NotificationHistory.reference.in_(references)) .filter(NotificationHistory.reference.in_(references))
.values(update_dict) .values(update_dict)
) )
updated_history_count = db.session.execute(stmt).scalar() or 0 result = db.session.execute(stmt)
updated_history_count = result.rowcount
return updated_count, updated_history_count return updated_count, updated_history_count