notifications-api/app/notifications/rest.py

from datetime import datetime

from flask import (
    Blueprint,
    jsonify,
    request,
    current_app
)

from app import api_user, encryption, create_uuid
from app.dao import (
    templates_dao,
    services_dao,
    notifications_dao,
    jobs_dao
)
from app.schemas import (
    email_notification_schema,
    sms_template_notification_schema,
    notification_status_schema,
    job_sms_template_notification_schema,
    job_email_template_notification_schema
)
from app.celery.tasks import send_sms, send_email
from sqlalchemy.orm.exc import NoResultFound

notifications = Blueprint('notifications', __name__)

from app.errors import register_errors

register_errors(notifications)

SMS_NOTIFICATION = 'sms'
EMAIL_NOTIFICATION = 'email'


@notifications.route('/notifications/<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('/notifications/sms', methods=['POST'])
def create_sms_notification():
    return send_notification(notification_type=SMS_NOTIFICATION, expects_job=False)


@notifications.route('/notifications/sms/service/<service_id>', methods=['POST'])
def create_sms_for_job(service_id):
    return send_notification(service_id=service_id, notification_type=SMS_NOTIFICATION, expects_job=True)


@notifications.route('/notifications/email', methods=['POST'])
def create_email_notification():
    return send_notification(notification_type=EMAIL_NOTIFICATION, expects_job=False)


@notifications.route('/notifications/email/service/<service_id>', methods=['POST'])
def create_email_notification_for_job(service_id):
    return send_notification(service_id=service_id, notification_type=EMAIL_NOTIFICATION, expects_job=True)


def send_notification(notification_type, service_id=None, expects_job=False):
    assert notification_type

    if not service_id:
        service_id = api_user['client']

    if expects_job:
        schema = job_sms_template_notification_schema if notification_type is SMS_NOTIFICATION else \
            job_email_template_notification_schema
    else:
        schema = sms_template_notification_schema if notification_type is SMS_NOTIFICATION else \
            email_notification_schema

    notification, errors = schema.load(request.get_json())
    if errors:
        return jsonify(result="error", message=errors), 400

    template = templates_dao.dao_get_template_by_id_and_service_id(
        template_id=notification['template'],
        service_id=service_id
    )

    if not template:
        return jsonify(
            result="error",
            message={
                'template': ['Template {} not found for service {}'.format(notification['template'], service_id)]
            }
        ), 400

    if expects_job:
        job = jobs_dao.dao_get_job_by_service_id_and_job_id(service_id, notification['job'])

        if not job:
            return jsonify(result="error", message={'job': ['Job {} not found'.format(notification['job'])]}), 400

    service = services_dao.dao_fetch_service_by_id(api_user['client'])

    if service.restricted:
        if notification_type is SMS_NOTIFICATION:
            if notification['to'] not in [user.mobile_number for user in service.users]:
                return jsonify(
                    result="error", message={'to': ['Invalid phone number for restricted service']}), 400
        else:
            if notification['to'] not in [user.email_address for user in service.users]:
                return jsonify(
                    result="error", message={'to': ['Email address not permitted for restricted service']}), 400

    notification_id = create_uuid()

    if notification_type is SMS_NOTIFICATION:
        send_sms.apply_async((
            service_id,
            notification_id,
            encryption.encrypt(notification),
            str(datetime.utcnow())),
            queue='sms')
    else:
        send_email.apply_async((
            service_id,
            notification_id,
            template.subject,
            "{}@{}".format(service.email_from, current_app.config['NOTIFY_EMAIL_DOMAIN']),
            encryption.encrypt(notification),
            str(datetime.utcnow())),
            queue='email')
    return jsonify({'notification_id': notification_id}), 201