mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-26 12:31:52 -05:00
4f33b6f406
Replace some 400s with more appropriate 500s. DAO methods that cause unexpected exceptions get caught and logged by errors.py 500 error handler.
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
import uuid
|
|
|
|
from flask import (
|
|
Blueprint,
|
|
jsonify,
|
|
request
|
|
)
|
|
|
|
from app import api_user, encryption
|
|
from app.aws_sqs import add_notification_to_queue
|
|
from app.dao import (templates_dao, notifications_dao)
|
|
from app.schemas import (
|
|
email_notification_schema,
|
|
sms_template_notification_schema,
|
|
notification_status_schema
|
|
)
|
|
from app.celery.tasks import send_sms
|
|
from sqlalchemy.orm.exc import NoResultFound
|
|
|
|
notifications = Blueprint('notifications', __name__)
|
|
|
|
from app.errors import register_errors
|
|
register_errors(notifications)
|
|
|
|
|
|
def create_notification_id():
|
|
return str(uuid.uuid4())
|
|
|
|
|
|
@notifications.route('/<string:notification_id>', methods=['GET'])
|
|
def get_notifications(notification_id):
|
|
try:
|
|
notification = notifications_dao.get_notification(api_user['client'], notification_id)
|
|
return jsonify({'notification': notification_status_schema.dump(notification).data}), 200
|
|
except NoResultFound:
|
|
return jsonify(result="error", message="not found"), 404
|
|
|
|
|
|
@notifications.route('/sms', methods=['POST'])
|
|
def create_sms_notification():
|
|
notification, errors = sms_template_notification_schema.load(request.get_json())
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
|
|
try:
|
|
templates_dao.get_model_templates(template_id=notification['template'], service_id=api_user['client'])
|
|
except NoResultFound:
|
|
return jsonify(result="error", message={'template': ['Template not found']}), 400
|
|
|
|
notification_id = create_notification_id()
|
|
|
|
send_sms.apply_async((
|
|
api_user['client'],
|
|
notification_id,
|
|
encryption.encrypt(notification)),
|
|
queue='sms')
|
|
return jsonify({'notification_id': notification_id}), 201
|
|
|
|
|
|
@notifications.route('/email', methods=['POST'])
|
|
def create_email_notification():
|
|
resp_json = request.get_json()
|
|
notification, errors = email_notification_schema.load(resp_json)
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
notification_id = add_notification_to_queue(api_user['client'], "admin", 'email', notification)
|
|
return jsonify({'notification_id': notification_id}), 201
|
|
|
|
|
|
@notifications.route('/sms/service/<service_id>', methods=['POST'])
|
|
def create_sms_for_service(service_id):
|
|
|
|
resp_json = request.get_json()
|
|
|
|
notification, errors = sms_template_notification_schema.load(resp_json)
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
|
|
template_id = notification['template']
|
|
job_id = notification['job']
|
|
|
|
# TODO: job/job_id is in notification and can used to update job status
|
|
|
|
# TODO: remove once beta is reading notifications from the queue
|
|
template = templates_dao.get_model_templates(template_id)
|
|
|
|
if template.service.id != uuid.UUID(service_id):
|
|
message = "Invalid template: id {} for service id: {}".format(template.id, service_id)
|
|
return jsonify(result="error", message=message), 400
|
|
|
|
notification_id = add_notification_to_queue(service_id, template_id, 'sms', notification)
|
|
return jsonify({'notification_id': notification_id}), 201
|