Moved the sending sms for a job into celery tasks

app/notifications/rest.py

@@ -8,17 +8,17 @@ from flask import (
 )
 
 from app import api_user, encryption
-from app.aws_sqs import add_notification_to_queue
 from app.dao import (
     templates_dao,
-    users_dao,
     services_dao,
-    notifications_dao
+    notifications_dao,
+    jobs_dao
 )
 from app.schemas import (
     email_notification_schema,
     sms_template_notification_schema,
-    notification_status_schema
+    notification_status_schema,
+    job_sms_template_notification_schema
 )
 from app.celery.tasks import send_sms, send_email
 from sqlalchemy.orm.exc import NoResultFound
@@ -73,6 +73,44 @@ def create_sms_notification():
     return jsonify({'notification_id': notification_id}), 201
 
 
+@notifications.route('/sms/service/<service_id>', methods=['POST'])
+def create_sms_for_service(service_id):
+    notification, errors = job_sms_template_notification_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 or service not found']}), 400
+
+    job_id = notification['job']
+
+    job = jobs_dao.get_job(service_id, job_id)
+
+    if not job:
+        return jsonify(result="error", message={'job': ['Job not found']}), 400
+
+    service = services_dao.dao_fetch_service_by_id(service_id)
+
+    if service.restricted:
+        if notification['to'] not in [user.email_address for user in service.users]:
+            return jsonify(result="error", message={'to': ['Invalid phone number for restricted service']}), 400
+
+    notification_id = create_notification_id()
+
+    send_sms.apply_async((
+        api_user['client'],
+        notification_id,
+        encryption.encrypt(notification),
+        job_id),
+        queue='sms')
+    return jsonify({'notification_id': notification_id}), 201
+
+
 @notifications.route('/email', methods=['POST'])
 def create_email_notification():
     notification, errors = email_notification_schema.load(request.get_json())
@@ -103,27 +141,3 @@ def create_email_notification():
         encryption.encrypt(notification)),
         queue='email')
     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