CON-72-story(notifications): admin receives notification when device is not seen in a while

- setup queue manager for when notifications are sent
- implement notifications when device is not seen in a while

[Delivers CON-72]

Author: [Arusey]

admin_notifications/helpers/create_notification.py

@@ -0,0 +1,17 @@
+from flask_socketio import send, emit
+from admin_notifications.models import AdminNotification
+from manage import socketio
+import celery
+
+
+@celery.task(name="create-notification")
+def create_notification(title, message, location_id):
+    notification = AdminNotification(
+        title=title,
+        message=message,
+        location_id=location_id,
+        status="unread"
+    )
+    notification.save()
+    new_notification = {"title": title, "message": message}
+    return socketio.emit('notification', {'notification': new_notification}, broadcast=True)  # noqa 501

admin_notifications/helpers/notification_templates.py

@@ -0,0 +1,6 @@
+def event_auto_cancelled_notification(event_name, room_name):
+    return {
+        "title": "Event Auto cancelled.",
+        "message": f"An event {event_name} in {room_name} \
+          has been auto cancelled."
+    }

admin_notifications/helpers/queue_manager.py

@@ -1,9 +1,7 @@
-from celery.schedules import crontab
-from cworker import celery_scheduler
-
-celery_scheduler.conf.beat_schedule = {
+from datetime import timedelta
+beat_schedule = {
     'run-check-device-last-seen-hourly': {
-        'task': 'check_device_last_seen',
-        'schedule': crontab(minute='*')
+        'task': 'check-device-last-seen',
+        'schedule': timedelta(seconds=5)
     }
 }

admin_notifications/socket_handler.py

@@ -0,0 +1,17 @@
+from flask_socketio import send
+from admin_notifications.models import AdminNotification
+
+
+def serialize_message(notification):
+    return {
+        "title": notification.title,
+        "message": notification.message,
+    }
+
+
+def send_notifications():
+    query = AdminNotification.query
+    notifications = query.filter_by(status="unread").all()
+    notifications = [serialize_message(notification)
+                     for notification in notifications]
+    return send(notifications, broadcast=True)

alembic/versions/1b53553ddacf_add_activity_column_to_devices_table.py

@@ -0,0 +1,34 @@
+"""add activity column to devices table
+
+Revision ID: 1b53553ddacf
+Revises: c65718cd4ed8
+Create Date: 2019-06-22 12:59:24.591339
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+
+# revision identifiers, used by Alembic.
+revision = '1b53553ddacf'
+down_revision = 'c65718cd4ed8'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    activitytype = postgresql.ENUM('online', 'offline', name='activitytype')
+    activitytype.create(op.get_bind())
+    op.add_column('devices', sa.Column('activity', sa.Enum(
+        'online', 'offline', name='activitytype'), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('devices', 'activity')
+    op.execute("DROP TYPE activitytype;")
+    # ### end Alembic commands ###

cworker.py

@@ -1,16 +1,26 @@
 import os
-
 from celery import Celery
 from app import create_app
-
+from admin_notifications.helpers.queue_manager import beat_schedule
 
 app = create_app(os.getenv('APP_SETTINGS') or 'default')
 app.app_context().push()
 
 
+app.config.update(
+    CELERY_BROKER_URL='redis://localhost:6379',
+    CELERY_RESULT_BACKEND='redis://localhost:6379',
+    CELERY_ACCEPT_CONTENT=['pickle'],
+    CELERYBEAT_SCHEDULE=beat_schedule
+)
+
+
 def make_celery(app):
-    celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'])
+    celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'], include=['admin_notifications.helpers.device_last_seen', 'admin_notifications.helpers.create_notification'], # noqa 501
+                    backend=app.config['CELERY_BROKER_URL'])
+
     celery.conf.update(app.config)
+    celery.conf.enable_utc = False
     TaskBase = celery.Task
 
     class ContextTask(TaskBase):

helpers/email/email_setup.py

@@ -30,7 +30,7 @@ def __init__(
             html=self.template,
             sender=self.sender)
 
-    @celery.task
+    @celery.task(name='asynchronous-email-notifications')
     def send_async_email(msg_dict):
         mail = Mail()
         msg = Message()

manage.py

@@ -3,13 +3,14 @@
 import bugsnag
 from flask_script import Manager, Shell
 from bugsnag.flask import handle_exceptions
+from flask_socketio import SocketIO
 
 
 # Configure bugnsag
 bugsnag.configure(
-  api_key=os.getenv('BUGSNAG_API_TOKEN'),
-  release_stage="development",
-  project_root="app"
+    api_key=os.getenv('BUGSNAG_API_TOKEN'),
+    release_stage="development",
+    project_root="app"
 )
 
 # local imports
@@ -18,6 +19,7 @@
 app = create_app(os.getenv('APP_SETTINGS') or 'default')
 handle_exceptions(app)
 manager = Manager(app)
+socketio = SocketIO(app)
 
 
 def make_shell_context():
@@ -30,4 +32,5 @@ def make_shell_context():
 
 
 if __name__ == '__main__':
+    socketio.run(app, debug=True)
     manager.run()

utilities/utility.py

@@ -89,6 +89,7 @@ class QuestionType(enum.Enum):
     missingitem = "missing_items"
 
 
+
 class StatusType(enum.Enum):
     read = "read"
     unread = "unread"