mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-14 01:02:09 -05:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import json
|
|
|
|
import boto3
|
|
from flask import (
|
|
Blueprint,
|
|
jsonify,
|
|
request,
|
|
current_app
|
|
)
|
|
from app import (notify_alpha_client, api_user)
|
|
from app.aws_sqs import add_notification_to_queue
|
|
from app.dao import (templates_dao, services_dao)
|
|
from app.schemas import (
|
|
email_notification_schema, sms_admin_notification_schema, sms_template_notification_schema)
|
|
import re
|
|
|
|
mobile_regex = re.compile("^\\+44[\\d]{10}$")
|
|
|
|
notifications = Blueprint('notifications', __name__)
|
|
|
|
|
|
@notifications.route('/<notification_id>', methods=['GET'])
|
|
def get_notifications(notification_id):
|
|
return jsonify(notify_alpha_client.fetch_notification_by_id(notification_id)), 200
|
|
|
|
|
|
@notifications.route('/sms', methods=['POST'])
|
|
def create_sms_notification():
|
|
resp_json = request.get_json()
|
|
|
|
# TODO: should create a different endpoint for the admin client to send verify codes.
|
|
if api_user['client'] == current_app.config.get('ADMIN_CLIENT_USER_NAME'):
|
|
notification, errors = sms_admin_notification_schema.load(resp_json)
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
template_id = 'admin'
|
|
message = notification['content']
|
|
else:
|
|
notification, errors = sms_template_notification_schema.load(resp_json)
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
template_id = notification['template']
|
|
message = notification['template']
|
|
|
|
add_notification_to_queue(api_user['client'], template_id, 'sms', notification)
|
|
return jsonify(notify_alpha_client.send_sms(
|
|
mobile_number=notification['to'], message=message)), 200
|
|
|
|
|
|
@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
|
|
# At the moment we haven't hooked up
|
|
# template handling for sending email notifications.
|
|
add_notification_to_queue(api_user['client'], "admin", 'email', notification)
|
|
return jsonify(notify_alpha_client.send_email(
|
|
notification['to_address'],
|
|
notification['body'],
|
|
notification['from_address'],
|
|
notification['subject']))
|