clean up and remove redundancy

app/celery/process_ses_receipts_tasks.py

@@ -1,10 +1,6 @@
-import enum
-import traceback
 from datetime import datetime, timedelta
-from json import decoder
 
 import iso8601
-import requests
 from celery.exceptions import Retry
 from flask import current_app, json
 from sqlalchemy.orm.exc import NoResultFound
@@ -25,7 +21,6 @@ from app.dao.service_callback_api_dao import (
     get_service_delivery_status_callback_api_for_service,
 )
 from app.models import NOTIFICATION_PENDING, NOTIFICATION_SENDING, Complaint
-from app.notifications.callbacks import create_complaint_callback_data
 
 
 @notify_celery.task(bind=True, name="process-ses-result", max_retries=5, default_retry_delay=300)

app/celery/service_callback_tasks.py

@@ -114,6 +114,7 @@ def create_delivery_status_callback_data(notification, service_callback_api):
         "notification_client_reference": notification.client_reference,
         "notification_to": notification.to,
         "notification_status": notification.status,
+        "notification_provider_response": notification.provider_response, # TODO do we have a test for provider_response
         "notification_created_at": notification.created_at.strftime(DATETIME_FORMAT),
         "notification_updated_at":
             notification.updated_at.strftime(DATETIME_FORMAT) if notification.updated_at else None,

app/notifications/callbacks.py

@@ -1,52 +0,0 @@
-from app.celery.service_callback_tasks import send_delivery_status_to_service
-from app.config import QueueNames
-from app.dao.service_callback_api_dao import (
-    get_service_delivery_status_callback_api_for_service,
-)
-
-
-def check_and_queue_callback_task(notification):
-    # queue callback task only if the service_callback_api exists
-    service_callback_api = get_service_delivery_status_callback_api_for_service(service_id=notification.service_id)
-    if service_callback_api:
-        notification_data = create_delivery_status_callback_data(notification, service_callback_api)
-
-        send_delivery_status_to_service.apply_async([str(notification.id), notification_data], queue=QueueNames.CALLBACKS)
-
-
-def create_delivery_status_callback_data(notification, service_callback_api):
-    from app import encryption
-    from app.utils import DATETIME_FORMAT
-
-    data = {
-        "notification_id": str(notification.id),
-        "notification_client_reference": notification.client_reference,
-        "notification_to": notification.to,
-        "notification_status": notification.status,
-        "notification_provider_response": notification.provider_response,
-        "notification_created_at": notification.created_at.strftime(DATETIME_FORMAT),
-        "notification_updated_at": notification.updated_at.strftime(DATETIME_FORMAT) if notification.updated_at else None,
-        "notification_sent_at": notification.sent_at.strftime(DATETIME_FORMAT) if notification.sent_at else None,
-        "notification_type": notification.notification_type,
-        "service_callback_api_url": service_callback_api.url,
-        "service_callback_api_bearer_token": service_callback_api.bearer_token,
-    }
-
-    return encryption.encrypt(data)
-
-
-def create_complaint_callback_data(complaint, notification, service_callback_api, recipient):
-    from app import encryption
-    from app.utils import DATETIME_FORMAT
-
-    data = {
-        "complaint_id": str(complaint.id),
-        "notification_id": str(notification.id),
-        "reference": notification.client_reference,
-        "to": recipient,
-        "complaint_date": complaint.complaint_date.strftime(DATETIME_FORMAT),
-        "service_callback_api_url": service_callback_api.url,
-        "service_callback_api_bearer_token": service_callback_api.bearer_token,
-    }
-
-    return encryption.encrypt(data)

app/notifications/notifications_ses_callback.py

@@ -1,25 +1,10 @@
-import enum
 from datetime import timedelta
 
-from flask import Blueprint, current_app, json, jsonify, request
+from flask import Blueprint, jsonify, request
 
 from app.celery.process_ses_receipts_tasks import process_ses_results
-from app.celery.service_callback_tasks import (
-    create_complaint_callback_data,
-    create_delivery_status_callback_data,
-    send_complaint_to_service,
-    send_delivery_status_to_service,
-)
 from app.config import QueueNames
-from app.dao.complaint_dao import save_complaint
-from app.dao.notifications_dao import dao_get_notification_history_by_reference
-from app.dao.service_callback_api_dao import (
-    get_service_complaint_callback_api_for_service,
-    get_service_delivery_status_callback_api_for_service,
-)
 from app.errors import InvalidRequest
-from app.models import Complaint
-from app.notifications.callbacks import create_complaint_callback_data
 from app.notifications.sns_handlers import sns_notification_handler
 
 ses_callback_blueprint = Blueprint('notifications_ses_callback', __name__)