feat(notifications): setup notific ations

- add new model for adminnotifications

[Delivers CON-71]

Author: codinger41

.coveragerc

@@ -0,0 +1,24 @@
+from flask_socketio import send
+from admin_notifications.models import AdminNotification
+from manage import socketio
+
+def create_notification(title, message, location_id):
+  notification = AdminNotification(
+    title=title,
+    message=message,
+    location_id=location_id,
+    status="unread"
+  )
+  notification.save()
+  
+  @socketio.on('notification')
+  def send_notification():
+    notification = { "title": title, "message": message }
+    return send(notification, broadcast=True)
+
+
+
+
+
+def create_notif_lol():
+  return create_notification(title="twre3r", message="rerf3ef3rf3rf", location_id=1)

notifications/__init__.py renamed to admin_notifications/__init__.py

@@ -0,0 +1,19 @@
+from sqlalchemy import (Column, String, Enum, Integer, ForeignKey)
+from helpers.database import Base
+from utilities.utility import Utility, StatusType
+
+
+class AdminNotification(Base, Utility):
+    __tablename__ = 'admin_notifications'
+
+    id = Column(Integer, primary_key=True) # noqa
+    title = Column(String, nullable=True)
+    message = Column(String, nullable=True)
+    date_received = Column(String, nullable=True)
+    date_read = Column(String, nullable=True)
+    status = Column(Enum(StatusType), default="unread")
+    location_id = Column(
+        Integer,
+        ForeignKey('locations.id', ondelete="CASCADE"),
+        nullable=True
+    )

notifications/helpers/__init__.py renamed to admin_notifications/helpers/__init__.py

@@ -0,0 +1,19 @@
+from flask_socketio import send
+from admin_notifications.models import AdminNotification
+from flask import jsonify
+
+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)
+
+def send_single_notification(title, message):
+  notification = { "title": title, "message": message }
+  return send(notification, broadcast=True)

admin_notifications/helpers/create_notification.py

@@ -45,7 +45,7 @@
 from api.response.models import Response
 from api.tag.models import Tag
 from api.structure.models import Structure
-
+from admin_notifications.models import AdminNotification
 
 target_metadata = Base.metadata
 

admin_notifications/models.py

@@ -0,0 +1,38 @@
+"""add admin notifications table
+
+Revision ID: c65718cd4ed8
+Revises: a5a4841351d7
+Create Date: 2019-06-21 08:38:07.110402
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'c65718cd4ed8'
+down_revision = 'a5a4841351d7'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('admin_notifications',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('title', sa.String(), nullable=True),
+    sa.Column('message', sa.String(), nullable=True),
+    sa.Column('date_received', sa.String(), nullable=True),
+    sa.Column('date_read', sa.String(), nullable=True),
+    sa.Column('status', sa.Enum('read', 'unread', name='statustype'), nullable=True),
+    sa.Column('location_id', sa.Integer(), nullable=True),
+    sa.ForeignKeyConstraint(['location_id'], ['locations.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('admin_notifications')
+    # ### end Alembic commands ###

admin_notifications/socket_handler.py

@@ -3,7 +3,6 @@
 from flask_graphql import GraphQLView
 from flask_cors import CORS
 from flask_json import FlaskJSON
-from flask_socketio import SocketIO
 
 from flask_mail import Mail
 from config import config
@@ -12,15 +11,13 @@
 from healthcheck_schema import healthcheck_schema
 from helpers.auth.authentication import Auth
 from api.analytics.analytics_request import AnalyticsRequest
-
 mail = Mail()
 
 
 def create_app(config_name):
     app = Flask(__name__)
     CORS(app)
     FlaskJSON(app)
-    SocketIO(app)
     app.config.from_object(config[config_name])
     config[config_name].init_app(app)
     mail.init_app(app)

alembic/env.py

@@ -3,7 +3,7 @@
 import bugsnag
 from flask_script import Manager, Shell
 from bugsnag.flask import handle_exceptions
-
+from flask_socketio import SocketIO
 
 # Configure bugnsag
 bugsnag.configure(
@@ -18,16 +18,23 @@
 app = create_app(os.getenv('APP_SETTINGS') or 'default')
 handle_exceptions(app)
 manager = Manager(app)
+socketio = SocketIO(app)
 
 
 def make_shell_context():
     return dict(app=app)
 
 
+@socketio.on('message')
+def hanldleMessage(msg):
+    from admin_notifications.socket_handler import send_notifications
+    return send_notifications()
+
+
 manager.add_command(
     "shell", Shell(
         make_context=make_shell_context))
 
-
 if __name__ == '__main__':
+    socketio.run(app, debug=True)
     manager.run()

alembic/versions/c65718cd4ed8_add_admin_notifications_table.py

@@ -16,14 +16,15 @@ coverage==4.5.1
 coveralls==1.8.0
 decorator==4.4.0
 docopt==0.6.2
+eventlet==0.25.0
 flake8==3.5.0
 Flask==0.12.2
 Flask-Cors==3.0.4
 Flask-GraphQL==1.4.1
 Flask-JSON==0.3.2
 Flask-Mail==0.9.1
 Flask-Script==2.0.6
-Flask-SocketIO==4.1.0
+flask-socketio==4.1.0
 Flask-Testing==0.7.1
 google-api-python-client==1.6.7
 graphene==2.1

app.py

@@ -86,3 +86,8 @@ class QuestionType(enum.Enum):
     rate = "rate"
     check = "check"
     textarea = "textarea"
+    missingitem = "missing_items"
+
+class StatusType(enum.Enum):
+    read = "read"
+    unread = "unread"