darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 07:35:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Celery task added

Browse Source
This commit is contained in:
Martyn Inglis
2016-02-15 11:11:20 +00:00
parent fcc1585fdf
commit 16c5e7bf10
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/notifications/rest.py
View File

@@ -11,6 +11,7 @@ from app.aws_sqs import add_notification_to_queue
from app.dao import (templates_dao) from app.dao import (templates_dao)
from app.schemas import ( from app.schemas import (
email_notification_schema, sms_template_notification_schema) email_notification_schema, sms_template_notification_schema)
from app import celery
notifications = Blueprint('notifications', __name__) notifications = Blueprint('notifications', __name__)
@@ -21,6 +22,14 @@ def get_notifications(notification_id):
return jsonify({'id': notification_id}), 200 return jsonify({'id': notification_id}), 200
@celery.task(name="make-sms", bind="True")
def send_sms(self):
print('Executing task id {0.id}, args: {0.args!r} kwargs: {0.kwargs!r}'.format(self.request))
from time import sleep
sleep(0.5)
print('finished')
@notifications.route('/sms', methods=['POST']) @notifications.route('/sms', methods=['POST'])
def create_sms_notification(): def create_sms_notification():
resp_json = request.get_json() resp_json = request.get_json()
@@ -29,6 +38,7 @@ def create_sms_notification():
if errors: if errors:
return jsonify(result="error", message=errors), 400 return jsonify(result="error", message=errors), 400
send_sms.delay()
add_notification_to_queue(api_user['client'], notification['template'], 'sms', notification) add_notification_to_queue(api_user['client'], notification['template'], 'sms', notification)
# TODO data to be returned # TODO data to be returned
return jsonify({}), 204 return jsonify({}), 204