darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-29 22:11:52 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use dao_update_notification to update billable_units for letter notifications

Browse Source
This commit is contained in:
Ken Tsang
2017-12-14 11:38:17 +00:00
parent f23074596b
commit 0045cd6b72
2 changed files with 6 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
app/celery/letters_pdf_tasks.py
View File

@@ -1,4 +1,3 @@
from datetime import datetime
from flask import current_app
import math
from requests import (
@@ -14,7 +13,7 @@ from app.config import QueueNames
from app.dao.notifications_dao import (
get_notification_by_id,
update_notification_status_by_id,
dao_update_notifications_by_reference
dao_update_notification
)
from app.statsd_decorators import statsd
@@ -35,22 +34,12 @@ def create_letters_pdf(self, notification_id):
notification.id, notification.reference, notification.created_at, len(pdf_data)))
s3.upload_letters_pdf(reference=notification.reference, crown=notification.service.crown, filedata=pdf_data)
updated_count = dao_update_notifications_by_reference(
references=[notification.reference],
update_dict={
"billable_units": billable_units,
"updated_at": datetime.utcnow()
}
)
notification.billable_units = billable_units
dao_update_notification(notification)
if not updated_count:
msg = "Update letter notification billing units failed: notification not found with reference {}".format(
notification.reference)
current_app.logger.error(msg)
else:
current_app.logger.info(
'Letter notification reference {reference}: billable units set to {billable_units}'.format(
reference=str(notification.reference), billable_units=billable_units))
current_app.logger.info(
'Letter notification reference {reference}: billable units set to {billable_units}'.format(
reference=str(notification.reference), billable_units=billable_units))
except (RequestException, BotoClientError):
try: