Dual Client setup

Author: devinmatte

config.env.py

@@ -48,3 +48,7 @@
 
 # Slack URL for pushing to #general
 SLACK_WEBHOOK_URL = environ.get("PACKET_SLACK_URL", None)
+
+# Packet Config
+PACKET_UPPER = environ.get("PACKET_UPPER", "packet.csh.rit.edu")
+PACKET_INTRO = environ.get("PACKET_INTRO", "freshmen-packet.csh.rit.edu")

packet/notifications.py

@@ -1,6 +1,6 @@
 import onesignal
 
-from packet import app, intro_onesignal_client
+from packet import app, intro_onesignal_client, csh_onesignal_client
 from packet.models import NotificationSubscription
 
 post_body = {
@@ -12,6 +12,18 @@
 }
 
 
+def send_notification(notification_body, subscriptions, client):
+    tokens = list(map(lambda subscription: subscription.token, subscriptions))
+    if tokens:
+        notification = onesignal.Notification(post_body=notification_body)
+        notification.post_body["include_player_ids"] = tokens
+        onesignal_response = client.send_notification(notification)
+        if onesignal_response.status_code == 200:
+            app.logger.info("The notification ({}) sent out successfully".format(notification.post_body))
+        else:
+            app.logger.warn("The notification ({}) was unsuccessful".format(notification.post_body))
+
+
 def packet_signed_notification(packet, signer):
     subscriptions = NotificationSubscription.query.filter_by(freshman_username=packet.freshman_username)
     if subscriptions:
@@ -22,25 +34,23 @@ def packet_signed_notification(packet, signer):
         notification.post_body["headings"]["en"] = 'New Packet Signature!'
         notification.post_body["chrome_web_icon"] = 'https://profiles.csh.rit.edu/image/' + signer
         notification.post_body["include_player_ids"] = tokens
+        notification.post_body["url"] = app.config["PROTOCOL"] + app.config["PACKET_INTRO"]
 
         onesignal_response = intro_onesignal_client.send_notification(notification)
         if onesignal_response.status_code == 200:
             app.logger.info("The notification ({}) sent out successfully".format(notification.post_body))
 
 
 def packet_100_percent_notification(packet):
-    # TODO: Split into csh and intro subscriptions
-    subscriptions = NotificationSubscription.query.all()
-    if subscriptions:
-        tokens = list(map(lambda subscription: subscription.token, subscriptions))
+    member_subscriptions = NotificationSubscription.query.filter(NotificationSubscription.member.isnot(None))
+    intro_subscriptions = NotificationSubscription.query.filter(NotificationSubscription.freshman_username.isnot(None))
 
-        notification = onesignal.Notification(post_body=post_body)
-        notification.post_body["contents"]["en"] = packet.freshman.name + ' got 💯 on packet!'
-        notification.post_body["headings"]["en"] = 'New 100% on Packet!'
+    if member_subscriptions or intro_subscriptions:
+        notification_body = post_body
+        notification_body["contents"]["en"] = packet.freshman.name + ' got 💯 on packet!'
+        notification_body["headings"]["en"] = 'New 100% on Packet!'
         # TODO: Issue #156
-        notification.post_body["chrome_web_icon"] = 'https://profiles.csh.rit.edu/image/' + packet.freshman_username
-        notification.post_body["include_player_ids"] = tokens
+        notification_body["chrome_web_icon"] = 'https://profiles.csh.rit.edu/image/' + packet.freshman_username
 
-        onesignal_response = onesignal_client.send_notification(notification)
-        if onesignal_response.status_code == 200:
-            app.logger.info("The notification ({}) sent out successfully".format(notification.post_body))
+        send_notification(notification_body, member_subscriptions, csh_onesignal_client)
+        send_notification(notification_body, intro_subscriptions, intro_onesignal_client)