darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-04 10:21:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix bug with created_at

Browse Source
This commit is contained in:
Kenneth Kehl
2025-01-13 14:21:43 -08:00
parent 4965bc2354
commit ba4301fc46
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/celery/scheduled_tasks.py
View File

@@ -315,6 +315,8 @@ def batch_insert_notifications(self):
notification_dict["status"] = notification_dict.pop("notification_status") notification_dict["status"] = notification_dict.pop("notification_status")
if not notification_dict.get("created_at"): if not notification_dict.get("created_at"):
notification_dict["created_at"] = utc_now() notification_dict["created_at"] = utc_now()
elif isinstance(notification_dict["created_at"], list):
notification_dict["created_at"] = notification_dict["created_at"][0]
notification = Notification(**notification_dict) notification = Notification(**notification_dict)
if notification is not None: if notification is not None:
batch.append(notification) batch.append(notification)
@@ -324,7 +326,10 @@ def batch_insert_notifications(self):
current_app.logger.exception("Notification batch insert failed") current_app.logger.exception("Notification batch insert failed")
for n in batch: for n in batch:
# Use 'created_at' as a TTL so we don't retry infinitely # Use 'created_at' as a TTL so we don't retry infinitely
if datetime.fromisoformat(n.created_at) < utc_now() - timedelta(seconds=50): notification_time = n.created_at
if isinstance(notification_time, str):
notification_time = datetime.fromisoformat(n.created_at)
if notification_time < utc_now() - timedelta(seconds=50):
current_app.logger.warning( current_app.logger.warning(
f"Abandoning stale data, could not write to db: {n.serialize_for_redis(n)}" f"Abandoning stale data, could not write to db: {n.serialize_for_redis(n)}"
) )