mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 23:41:17 -05:00
37a2dc7925
- this allows us to send a notification to a provider by means of an API call - This is in addition to the celery code. - idea is that we can use this method to help speed up throughput by generating API traffic by node/lambda etc to supplement the celery code in times of high load.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from flask import Blueprint, jsonify
|
|
|
|
from app.delivery import send_to_providers
|
|
from app.models import EMAIL_TYPE
|
|
from app.celery import provider_tasks
|
|
from app.dao import notifications_dao
|
|
from flask import current_app
|
|
|
|
delivery_blueprint = Blueprint('delivery', __name__)
|
|
|
|
from app.errors import register_errors
|
|
|
|
register_errors(delivery_blueprint)
|
|
|
|
|
|
@delivery_blueprint.route('/deliver/notification/<uuid:notification_id>', methods=['POST'])
|
|
def send_notification_to_provider(notification_id):
|
|
notification = notifications_dao.get_notification_by_id(notification_id)
|
|
if not notification:
|
|
return jsonify({"result": "error", "message": "No result found"}), 404
|
|
|
|
if notification.notification_type == EMAIL_TYPE:
|
|
send_response(
|
|
send_to_providers.send_email_to_provider,
|
|
provider_tasks.deliver_email,
|
|
notification,
|
|
'send-email')
|
|
else:
|
|
send_response(
|
|
send_to_providers.send_sms_to_provider,
|
|
provider_tasks.deliver_sms,
|
|
notification,
|
|
'send-sms')
|
|
return jsonify({}), 204
|
|
|
|
|
|
def send_response(send_call, task_call, notification, queue):
|
|
try:
|
|
send_call(notification)
|
|
except Exception as e:
|
|
current_app.logger.exception(
|
|
"Failed to send notification, retrying in celery. ID {} type {}".format(
|
|
notification.id,
|
|
notification.notification_type),
|
|
e)
|
|
task_call.apply_async((str(notification.id)), queue=queue)
|