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

add a test

Browse Source
This commit is contained in:
Kenneth Kehl
2025-01-13 10:00:18 -08:00
parent 28470468e2
commit a92eb91470
3 changed files with 48 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/dao/notifications_dao.py
View File

@@ -2,7 +2,6 @@ import json
from datetime import timedelta
from time import time
import sqlalchemy
from flask import current_app
from sqlalchemy import (
TIMESTAMP,
@@ -803,11 +802,8 @@ def dao_close_out_delivery_receipts():
def dao_batch_insert_notifications(batch):
current_app.logger.info("DOING BATCH INSERT IN DAO")
try:
db.session.bulk_save_objects(batch)
db.session.commit()
current_app.logger.info(f"SUCCESSFULLY INSERTED: {len(batch)}")
return len(batch)
except sqlalchemy.exc.SQLAlchemyError as e:
current_app.logger.exception(f"Error during batch insert {e}")
db.session.bulk_save_objects(batch)
db.session.commit()
current_app.logger.info(f"Batch inserted notifications: {len(batch)}")
return len(batch)