Adding model to collect subscriptions

Author: devinmatte

migrations/versions/53768f0a4850_notifications.py

@@ -0,0 +1,35 @@
+"""notifications
+
+Revision ID: 53768f0a4850
+Revises: eecf30892d0e
+Create Date: 2019-08-06 22:15:04.400982
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '53768f0a4850'
+down_revision = 'eecf30892d0e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('notification_subscriptions',
+    sa.Column('member', sa.String(length=36), nullable=True),
+    sa.Column('freshman_username', sa.String(length=10), nullable=True),
+    sa.Column('token', sa.String(length=256), nullable=False),
+    sa.ForeignKeyConstraint(['freshman_username'], ['freshman.rit_username'], ),
+    sa.PrimaryKeyConstraint('token')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('notification_subscriptions')
+    # ### end Alembic commands ###
+

packet/models.py

@@ -154,3 +154,10 @@ class MiscSignature(db.Model):
     updated = Column(DateTime, default=datetime.now, onupdate=datetime.now, nullable=False)
 
     packet = relationship("Packet", back_populates="misc_signatures")
+
+
+class NotificationSubscription(db.Model):
+    __tablename__ = "notification_subscriptions"
+    member = Column(String(36), nullable=True)
+    freshman_username = Column(ForeignKey("freshman.rit_username"), nullable=True)
+    token = Column(String(256), primary_key=True, nullable=False)

packet/routes/api.py

@@ -7,7 +7,7 @@
 from packet.context_processors import get_rit_name
 from packet.mail import send_report_mail
 from packet.utils import before_request, packet_auth, notify_slack
-from packet.models import Packet, MiscSignature
+from packet.models import Packet, MiscSignature, NotificationSubscription
 
 
 @app.route("/api/v1/sign/<packet_id>/", methods=["POST"])
@@ -40,6 +40,20 @@ def sign(packet_id, info):
     return "Error: Signature not valid.  Reason: Unknown"
 
 
+@app.route("/api/v1/subscribe/", methods=["POST"])
+@packet_auth
+@before_request
+def subscribe(info):
+    data = request.form
+    if app.config["REALM"] == "csh":
+        subscription = NotificationSubscription(token=data['token'], member=info["uid"])
+    else:
+        subscription = NotificationSubscription(token=data['token'], freshman_username=info["uid"])
+    db.session.add(subscription)
+    db.session.commit()
+    return "Token subscribed for " + info["uid"]
+
+
 @app.route("/api/v1/report/", methods=["POST"])
 @packet_auth
 @before_request

packet/routes/shared.py

@@ -80,4 +80,3 @@ def service_worker():
 @app.route('/OneSignalSDKUpdaterWorker.js', methods=['GET'])
 def update_service_worker():
     return app.send_static_file('js/update-sw.js')
-

packet/templates/include/head.html

@@ -47,8 +47,16 @@
             OneSignal.showNativePrompt();
             OneSignal.on("subscriptionChange", function(){
 		OneSignal.getUserId().then(function(result){
-			// TODO: Send to backend
-			console.log(result);
+			$.ajax({
+            			url: "/api/v1/subscribe/",
+            			method: "POST",
+            			data: {
+                			token: result
+            			},
+            			success: function (data) {
+                			console.log(data);
+            			}
+        		});
 		});
 	    });
         });