From 8315c52a103f3ae461a8acca21cb5666ff22e716 Mon Sep 17 00:00:00 2001 From: Vincent Cai <72774400+vcai122@users.noreply.github.com> Date: Sun, 17 Nov 2024 12:09:06 -0500 Subject: [PATCH 1/2] Revamp Notif Logic (#324) * Rewrite notif logic according to new specs * fix tests and comment out a lot --- backend/tests/user/test_notifs.py | 436 ++++++------------ backend/user/admin.py | 12 +- ..._remove_notificationtoken_user_and_more.py | 64 +++ backend/user/models.py | 63 +-- backend/user/notifications.py | 65 +-- backend/user/serializers.py | 35 +- backend/user/urls.py | 19 +- backend/user/views.py | 154 ++++--- k8s/main.ts | 14 +- 9 files changed, 377 insertions(+), 485 deletions(-) create mode 100644 backend/user/migrations/0010_remove_notificationtoken_user_and_more.py diff --git a/backend/tests/user/test_notifs.py b/backend/tests/user/test_notifs.py index cbeb8075..6fa67b79 100644 --- a/backend/tests/user/test_notifs.py +++ b/backend/tests/user/test_notifs.py @@ -1,16 +1,12 @@ -import datetime import json from unittest import mock from django.contrib.auth import get_user_model -from django.core.management import call_command -from django.test import TestCase -from django.utils import timezone +from django.test import TestCase, TransactionTestCase from identity.identity import attest, container, get_platform_jwks from rest_framework.test import APIClient -from gsr_booking.models import GSR, GSRBooking, Reservation -from user.models import NotificationSetting, NotificationToken +from user.models import IOSNotificationToken, NotificationService User = get_user_model() @@ -42,173 +38,92 @@ def mock_client(is_dev): return MockAPNsClient() -class TestNotificationToken(TestCase): - """Tests for CRUD Notification Tokens""" +class TestIOSNotificationToken(TestCase): + """Tests for associating and deleting IOS Notification Tokens""" def setUp(self): self.client = APIClient() self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") self.client.force_authenticate(user=self.test_user) - - def test_post_save(self): - # asserts that post save hook in creating tokens works correctly - self.assertEqual(1, NotificationToken.objects.all().count()) - self.assertEqual(self.test_user, NotificationToken.objects.all().first().user) - - def test_create_update_token(self): - NotificationToken.objects.all().delete() - - # test that creating token returns correct response - payload = {"kind": "IOS", "token": "test123"} - response = self.client.post("/user/notifications/tokens/", payload) - res_json = json.loads(response.content) - self.assertEqual("IOS", res_json["kind"]) - self.assertEqual("test123", res_json["token"]) - self.assertEqual(3, len(res_json)) - self.assertEqual(1, NotificationToken.objects.all().count()) - - # update token - new_payload = {"kind": "IOS", "token": "newtoken"} - response = self.client.patch(f"/user/notifications/tokens/{res_json['id']}/", new_payload) - res_json = json.loads(response.content) - self.assertEqual("newtoken", res_json["token"]) - self.assertEqual(1, NotificationToken.objects.all().count()) - - def test_create_token_again_fail(self): - # test that creating token returns correct response - payload = {"kind": "IOS", "token": "test123"} - response = self.client.post("/user/notifications/tokens/", payload) - self.assertEqual(response.status_code, 400) - - def test_get_token(self): - NotificationToken.objects.all().delete() - - # create token - payload = {"kind": "IOS", "token": "test123"} - response = self.client.post("/user/notifications/tokens/", payload) - - response = self.client.get("/user/notifications/tokens/") - res_json = json.loads(response.content) - self.assertEqual("IOS", res_json[0]["kind"]) - self.assertEqual("test123", res_json[0]["token"]) - self.assertEqual(1, len(res_json)) - self.assertEqual(3, len(res_json[0])) - self.assertEqual(1, NotificationToken.objects.all().count()) - - -class TestNotificationSetting(TestCase): + self.token = "1234" + + def test_create_token(self): + response = self.client.post(f"/user/notifications/tokens/ios/{self.token}/") + self.assertEqual(201, response.status_code) + self.assertEqual(1, IOSNotificationToken.objects.all().count()) + the_token = IOSNotificationToken.objects.first() + self.assertEqual(self.token, the_token.token) + self.assertEqual(self.test_user, the_token.user) + self.assertEqual(False, the_token.is_dev) + + def test_update_token(self): + # test that posting to same token updates the token + user2 = User.objects.create_user("user2", "user2@seas.upenn.edu", "user2") + IOSNotificationToken.objects.create(user=user2, token=self.token, is_dev=True) + + response = self.client.post(f"/user/notifications/tokens/ios/{self.token}/") + self.assertEqual(200, response.status_code) + self.assertEqual(1, IOSNotificationToken.objects.all().count()) + the_token = IOSNotificationToken.objects.first() + self.assertEqual(self.token, the_token.token) + self.assertEqual(self.test_user, the_token.user) + self.assertEqual(False, the_token.is_dev) + + def test_delete_token(self): + response = self.client.post(f"/user/notifications/tokens/ios/{self.token}/") + self.assertEqual(201, response.status_code) + self.assertEqual(1, IOSNotificationToken.objects.all().count()) + + response = self.client.delete(f"/user/notifications/tokens/ios/{self.token}/") + self.assertEqual(200, response.status_code) + self.assertEqual(0, IOSNotificationToken.objects.all().count()) + + +class TestNotificationService(TransactionTestCase): """Tests for CRUD Notification Settings""" def setUp(self): + NotificationService.objects.bulk_create( + [ + NotificationService(name="PENN_MOBILE"), + NotificationService(name="OHQ"), + ] + ) self.client = APIClient() self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") self.client.force_authenticate(user=self.test_user) - initialize_b2b() - - def test_get_settings(self): - # test that settings visible via GET - response = self.client.get("/user/notifications/settings/") - res_json = json.loads(response.content) - self.assertEqual(len(NotificationSetting.SERVICE_OPTIONS), len(res_json)) - for setting in res_json: - self.assertFalse(setting["enabled"]) - def test_invalid_settings_update(self): - NotificationToken.objects.all().delete() - payload = {"kind": "IOS", "token": "test123"} - response = self.client.post("/user/notifications/tokens/", payload) + def test_get_services(self): + # test that services visible via GET + response = self.client.get("/user/notifications/services/") res_json = json.loads(response.content) + self.assertEqual(["OHQ", "PENN_MOBILE"], sorted(res_json)) - response = self.client.get("/user/notifications/settings/PENN_MOBILE/check/") - res_json = json.loads(response.content) - settings_id = res_json["id"] - payload = {"service": "PENN_MOBILE", "enabled": True} - response = self.client.patch(f"/user/notifications/settings/{settings_id}/", payload) - res_json = json.loads(response.content) - self.assertEqual(res_json["service"], "PENN_MOBILE") - self.assertTrue(res_json["enabled"]) - - def test_valid_settings_update(self): - NotificationToken.objects.all().delete() - response = self.client.get("/user/notifications/settings/PENN_MOBILE/check/") - res_json = json.loads(response.content) - self.assertFalse(res_json["enabled"]) - - payload = {"kind": "IOS", "token": "test123"} - response = self.client.post("/user/notifications/tokens/", payload) - res_json = json.loads(response.content) - - response = self.client.get("/user/notifications/settings/PENN_MOBILE/check/") - res_json = json.loads(response.content) - settings_id = res_json["id"] - payload = {"service": "OHQ", "enabled": True} - response = self.client.patch(f"/user/notifications/settings/{settings_id}/", payload) - self.assertEqual(response.status_code, 400) - - def test_create_update_check_settings(self): - # test that invalid settings are rejected - NotificationSetting.objects.filter(service="PENN_MOBILE").delete() - payload = {"service": "Penn Mobile", "enabled": True} - response = self.client.post("/user/notifications/settings/", payload) - res_json = json.loads(response.content) - self.assertNotEqual(res_json, payload) - - # test that settings can be created - payload = {"service": "PENN_MOBILE", "enabled": True} - response = self.client.post("/user/notifications/settings/", payload) - res_json = json.loads(response.content) - self.assertEqual(res_json["service"], "PENN_MOBILE") - self.assertTrue(res_json["enabled"]) - - # test fail of re-creating settings - response = self.client.post("/user/notifications/settings/", payload) - self.assertEqual(response.status_code, 400) - - # since token empty, should still return false - response = self.client.get("/user/notifications/tokens/") + def test_get_settings(self): + response = self.client.get("/user/notifications/settings/") res_json = json.loads(response.content) - token_id = res_json[0]["id"] + self.assertDictEqual({"PENN_MOBILE": False, "OHQ": False}, res_json) - # update token to nonempty value - payload = {"kind": "IOS", "token": "test123"} - response = self.client.put(f"/user/notifications/tokens/{token_id}/", payload) - res_json = json.loads(response.content) - self.assertEqual("test123", res_json["token"]) - self.assertEqual(1, NotificationToken.objects.all().count()) + def test_update_settings(self): + response = self.client.put( + "/user/notifications/settings/", + json.dumps({"PENN_MOBILE": True}), + content_type="application/json", + ) + self.assertEqual(200, response.status_code) - # re-request check - response = self.client.get("/user/notifications/settings/PENN_MOBILE/check/") + response = self.client.get("/user/notifications/settings/") res_json = json.loads(response.content) - self.assertTrue(res_json["enabled"]) + self.assertDictEqual({"PENN_MOBILE": True, "OHQ": False}, res_json) - def test_check_fail(self): - # since invalid setting, should return error - response = self.client.get("/user/notifications/settings/PENN_MOBIL/check/") - self.assertEqual(response.status_code, 400) - - # def test_b2b_queryset_empty(self): - # self.client.logout() - # b2b_client = get_b2b_client() - # response = b2b_client.get("/user/notifications/settings/") - # self.assertEqual(response.status_code, 200) - # res_json = json.loads(response.content) - # self.assertEqual(0, len(res_json)) - - # def test_b2b_check(self): - # self.client.logout() - # b2b_client = get_b2b_client() - # response = b2b_client.get( - # "/user/notifications/settings/PENN_MOBILE/check/?pennkey=user" - # ) - # self.assertEqual(response.status_code, 200) - # res_json = json.loads(response.content) - # self.assertEqual(res_json["service"], "PENN_MOBILE") - # self.assertFalse(res_json["enabled"]) - - def test_b2b_auth_fails(self): - self.client.logout() - response = self.client.get("/user/notifications/settings/PENN_MOBILE/check/?pennkey=user") - self.assertEqual(response.status_code, 403) + def test_invalid_settings_update(self): + # Requires TransactionTestCase since relies on database rollback + response = self.client.put( + "/user/notifications/settings/", + json.dumps({"UNKNOWN": True, "sPENN_MOBILE": True, "ABC": True, "OHQ": True}), + content_type="application/json", + ) + self.assertEqual(400, response.status_code) class TestNotificationAlert(TestCase): @@ -217,31 +132,26 @@ class TestNotificationAlert(TestCase): def setUp(self): self.client = APIClient() - # create user1 - self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") - self.client.force_authenticate(user=self.test_user) - token_obj = NotificationToken.objects.get(user=self.test_user) - token_obj.token = "test123" - token_obj.save() + NotificationService.objects.bulk_create( + [ + NotificationService(name="PENN_MOBILE"), + NotificationService(name="OHQ"), + ] + ) - # create user2 - self.test_user = User.objects.create_user("user2", "user2@seas.upenn.edu", "user2") - self.client.force_authenticate(user=self.test_user) - token_obj = NotificationToken.objects.get(user=self.test_user) - token_obj.token = "test234" - token_obj.save() - setting = NotificationSetting.objects.get(token=token_obj, service="PENN_MOBILE") - setting.enabled = True - setting.save() + user1 = User.objects.create_user("user", "user@seas.upenn.edu", "user") + IOSNotificationToken.objects.create(user=user1, token="test123") + + user2 = User.objects.create_user("user2", "user2@seas.upenn.edu", "user2") + IOSNotificationToken.objects.create(user=user2, token="test234") + user2.notificationservice_set.add("PENN_MOBILE") # create user3 user3 = User.objects.create_user("user3", "user3@seas.upenn.edu", "user3") - token_obj = NotificationToken.objects.get(user=user3) - token_obj.token = "test234" - token_obj.save() - setting = NotificationSetting.objects.get(token=token_obj, service="PENN_MOBILE") - setting.enabled = True - setting.save() + IOSNotificationToken.objects.create(user=user3, token="test345") + user3.notificationservice_set.add("PENN_MOBILE") + + self.client.force_authenticate(user=user3) initialize_b2b() @@ -277,112 +187,72 @@ def test_single_notif(self): self.assertEqual(1, len(res_json["success_users"])) self.assertEqual(0, len(res_json["failed_users"])) - @mock.patch("user.notifications.get_client", mock_client) - def test_batch_notif(self): - # update all settings to be enabled - NotificationSetting.objects.all().update(enabled=True) - - # test notif - payload = { - "users": ["user2", "user1", "user3"], - "title": "Test", - "body": ":D", - "service": "PENN_MOBILE", - } - response = self.client.post( - "/user/notifications/alerts/", json.dumps(payload), content_type="application/json" - ) - res_json = json.loads(response.content) - self.assertEqual(1, len(res_json["success_users"])) - self.assertEqual(0, len(res_json["failed_users"])) - - # @mock.patch("user.notifications.get_client", mock_client) - # def test_b2b_batch_alert(self): - # self.client.logout() - # b2b_client = get_b2b_client() - # payload = { - # "users": ["user", "user2", "user3"], - # "title": "Test", - # "body": ":D", - # "service": "PENN_MOBILE", - # } - # response = b2b_client.post( - # "/user/notifications/alerts/", - # json.dumps(payload), - # content_type="application/json", - # ) - # res_json = json.loads(response.content) - # self.assertEqual(2, len(res_json["success_users"])) - # self.assertEqual(1, len(res_json["failed_users"])) - - -class TestSendGSRReminders(TestCase): - """Test Sending GSR Reminders""" - - def setUp(self): - call_command("load_gsrs") - self.client = APIClient() - self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") - self.client.force_authenticate(user=self.test_user) - - # enabling tokens and settings - token_obj = NotificationToken.objects.get(user=self.test_user) - token_obj.token = "test123" - token_obj.save() - - setting = NotificationSetting.objects.get( - token=token_obj, service=NotificationSetting.SERVICE_GSR_BOOKING - ) - setting.enabled = True - setting.save() - - # creating reservation and booking for notifs - g = GSRBooking.objects.create( - user=self.test_user, - gsr=GSR.objects.all().first(), - room_id=1, - room_name="Room", - start=timezone.now() + datetime.timedelta(minutes=5), - end=timezone.now() + datetime.timedelta(minutes=35), - ) - - r = Reservation.objects.create( - start=g.start, - end=g.end, - creator=self.test_user, - ) - - g.reservation = r - g.save() - - @mock.patch("user.notifications.get_client", mock_client) - def test_send_reminder(self): - call_command("send_gsr_reminders") - r = Reservation.objects.all().first() - self.assertTrue(r.reminder_sent) - - def test_send_reminder_no_gsrs(self): - GSRBooking.objects.all().delete() - call_command("send_gsr_reminders") - r = Reservation.objects.all().first() - self.assertFalse(r.reminder_sent) - - -class TestSendShadowNotifs(TestCase): - """Test Sending Shadow Notifications""" - - def setUp(self): - self.client = APIClient() - self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") - self.client.force_authenticate(user=self.test_user) - token_obj = NotificationToken.objects.get(user=self.test_user) - token_obj.token = "test123" - token_obj.save() - - @mock.patch("user.notifications.get_client", mock_client) - def test_shadow_notifications(self): - # call command on every user - call_command("send_shadow_notifs", "yes", '{"test":"test"}') - # call command on specific set of users - call_command("send_shadow_notifs", "no", '{"test":"test"}', users="user1") +# TODO: FIX IN LATER PR + +# class TestSendGSRReminders(TestCase): +# """Test Sending GSR Reminders""" + +# def setUp(self): +# call_command("load_gsrs") +# self.client = APIClient() +# user = User.objects.create_user("user", "user@seas.upenn.edu", "user") +# user.iosnotificationtoken_set.create(token="test123") + + +# setting = NotificationSetting.objects.get( +# token=token_obj, service=NotificationSetting.SERVICE_GSR_BOOKING +# ) +# setting.enabled = True +# setting.save() + +# # creating reservation and booking for notifs +# g = GSRBooking.objects.create( +# user=self.test_user, +# gsr=GSR.objects.all().first(), +# room_id=1, +# room_name="Room", +# start=timezone.now() + datetime.timedelta(minutes=5), +# end=timezone.now() + datetime.timedelta(minutes=35), +# ) + +# r = Reservation.objects.create( +# start=g.start, +# end=g.end, +# creator=self.test_user, +# ) + +# g.reservation = r +# g.save() + +# @mock.patch("user.notifications.get_client", mock_client) +# def test_send_reminder(self): +# call_command("send_gsr_reminders") +# r = Reservation.objects.all().first() +# self.assertTrue(r.reminder_sent) + +# def test_send_reminder_no_gsrs(self): +# GSRBooking.objects.all().delete() +# call_command("send_gsr_reminders") +# r = Reservation.objects.all().first() +# self.assertFalse(r.reminder_sent) + + +# class TestSendShadowNotifs(TestCase): +# """Test Sending Shadow Notifications""" + +# def setUp(self): +# self.client = APIClient() +# self.test_user = User.objects.create_user("user", "user@seas.upenn.edu", "user") +# self.client.force_authenticate(user=self.test_user) +# token_obj = IOSNotificationToken.objects.get(user=self.test_user) +# token_obj.token = "test123" +# token_obj.save() + +# @mock.patch("user.notifications.get_client", mock_client) +# def test_shadow_notifications(self): +# # call command on every user +# call_command("send_shadow_notifs", "yes", '{"test":"test"}') + +# # call command on specific set of users +# call_command("send_shadow_notifs", "no", '{"test":"test"}', users="user1") diff --git a/backend/user/admin.py b/backend/user/admin.py index 75fe9c36..2858523a 100644 --- a/backend/user/admin.py +++ b/backend/user/admin.py @@ -1,8 +1,14 @@ from django.contrib import admin -from user.models import NotificationSetting, NotificationToken, Profile +from user.models import AndroidNotificationToken, IOSNotificationToken, NotificationService, Profile -admin.site.register(NotificationToken) -admin.site.register(NotificationSetting) +# custom IOSNotificationToken admin +class IOSNotificationTokenAdmin(admin.ModelAdmin): + list_display = ("token", "user", "is_dev") + + +admin.site.register(IOSNotificationToken, IOSNotificationTokenAdmin) +admin.site.register(AndroidNotificationToken) +admin.site.register(NotificationService) admin.site.register(Profile) diff --git a/backend/user/migrations/0010_remove_notificationtoken_user_and_more.py b/backend/user/migrations/0010_remove_notificationtoken_user_and_more.py new file mode 100644 index 00000000..0d9f3b30 --- /dev/null +++ b/backend/user/migrations/0010_remove_notificationtoken_user_and_more.py @@ -0,0 +1,64 @@ +# Generated by Django 5.0.2 on 2024-11-11 05:24 + +import django.db.models.deletion +from django.conf import settings +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("user", "0009_profile_fitness_preferences"), + migrations.swappable_dependency(settings.AUTH_USER_MODEL), + ] + + operations = [ + migrations.RemoveField( + model_name="notificationtoken", + name="user", + ), + migrations.CreateModel( + name="AndroidNotificationToken", + fields=[ + ("token", models.CharField(max_length=255, primary_key=True, serialize=False)), + ( + "user", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL + ), + ), + ], + options={ + "abstract": False, + }, + ), + migrations.CreateModel( + name="IOSNotificationToken", + fields=[ + ("token", models.CharField(max_length=255, primary_key=True, serialize=False)), + ("is_dev", models.BooleanField(default=False)), + ( + "user", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL + ), + ), + ], + options={ + "abstract": False, + }, + ), + migrations.CreateModel( + name="NotificationService", + fields=[ + ("name", models.CharField(max_length=255, primary_key=True, serialize=False)), + ("enabled_users", models.ManyToManyField(blank=True, to=settings.AUTH_USER_MODEL)), + ], + ), + migrations.DeleteModel( + name="NotificationSetting", + ), + migrations.DeleteModel( + name="NotificationToken", + ), + ] diff --git a/backend/user/models.py b/backend/user/models.py index dc71b5c3..aa113389 100644 --- a/backend/user/models.py +++ b/backend/user/models.py @@ -11,46 +11,24 @@ class NotificationToken(models.Model): - KIND_IOS = "IOS" - KIND_ANDROID = "ANDROID" - KIND_OPTIONS = ((KIND_IOS, "iOS"), (KIND_ANDROID, "Android")) + user = models.ForeignKey(User, on_delete=models.CASCADE) + token = models.CharField(max_length=255, primary_key=True) + + class Meta: + abstract = True + + +class IOSNotificationToken(NotificationToken): + is_dev = models.BooleanField(default=False) - user = models.OneToOneField(User, on_delete=models.CASCADE) - kind = models.CharField(max_length=7, choices=KIND_OPTIONS, default=KIND_IOS) - token = models.CharField(max_length=255) - - -class NotificationSetting(models.Model): - SERVICE_CFA = "CFA" - SERVICE_PENN_CLUBS = "PENN_CLUBS" - SERVICE_PENN_BASICS = "PENN_BASICS" - SERVICE_OHQ = "OHQ" - SERVICE_PENN_COURSE_ALERT = "PENN_COURSE_ALERT" - SERVICE_PENN_COURSE_PLAN = "PENN_COURSE_PLAN" - SERVICE_PENN_COURSE_REVIEW = "PENN_COURSE_REVIEW" - SERVICE_PENN_MOBILE = "PENN_MOBILE" - SERVICE_GSR_BOOKING = "GSR_BOOKING" - SERVICE_DINING = "DINING" - SERVICE_UNIVERSITY = "UNIVERSITY" - SERVICE_LAUNDRY = "LAUNDRY" - SERVICE_OPTIONS = ( - (SERVICE_CFA, "CFA"), - (SERVICE_PENN_CLUBS, "Penn Clubs"), - (SERVICE_PENN_BASICS, "Penn Basics"), - (SERVICE_OHQ, "OHQ"), - (SERVICE_PENN_COURSE_ALERT, "Penn Course Alert"), - (SERVICE_PENN_COURSE_PLAN, "Penn Course Plan"), - (SERVICE_PENN_COURSE_REVIEW, "Penn Course Review"), - (SERVICE_PENN_MOBILE, "Penn Mobile"), - (SERVICE_GSR_BOOKING, "GSR Booking"), - (SERVICE_DINING, "Dining"), - (SERVICE_UNIVERSITY, "University"), - (SERVICE_LAUNDRY, "Laundry"), - ) - - token = models.ForeignKey(NotificationToken, on_delete=models.CASCADE) - service = models.CharField(max_length=30, choices=SERVICE_OPTIONS, default=SERVICE_PENN_MOBILE) - enabled = models.BooleanField(default=True) + +class AndroidNotificationToken(NotificationToken): + pass + + +class NotificationService(models.Model): + name = models.CharField(max_length=255, primary_key=True) + enabled_users = models.ManyToManyField(User, blank=True) class Profile(models.Model): @@ -70,10 +48,3 @@ def create_or_update_user_profile(sender, instance, created, **kwargs): object exists for that User, it will create one """ Profile.objects.get_or_create(user=instance) - - # notifications - token, _ = NotificationToken.objects.get_or_create(user=instance) - for service, _ in NotificationSetting.SERVICE_OPTIONS: - setting = NotificationSetting.objects.filter(token=token, service=service).first() - if not setting: - NotificationSetting.objects.create(token=token, service=service, enabled=False) diff --git a/backend/user/notifications.py b/backend/user/notifications.py index 3cc73547..e4ffa033 100644 --- a/backend/user/notifications.py +++ b/backend/user/notifications.py @@ -19,23 +19,20 @@ collections.MutableMapping = abc.MutableMapping from apns2.client import APNsClient -from apns2.credentials import TokenCredentials from apns2.payload import Payload from celery import shared_task -from user.models import NotificationToken - # taken from the apns2 method for batch notifications Notification = collections.namedtuple("Notification", ["token", "payload"]) -def send_push_notifications(users, service, title, body, delay=0, is_dev=False, is_shadow=False): +def send_push_notifications(tokens, category, title, body, delay=0, is_dev=False, is_shadow=False): """ Sends push notifications. - :param users: list of usernames to send notifications to or 'None' if to all - :param service: service to send notifications for or 'None' if ignoring settings + :param tokens: nonempty list of tokens to send notifications to + :param category: category to send notifications for :param title: title of notification :param body: body of notification :param delay: delay in seconds before sending notification @@ -43,37 +40,15 @@ def send_push_notifications(users, service, title, body, delay=0, is_dev=False, :return: tuple of (list of success usernames, list of failed usernames) """ - # collect available usernames & their respective device tokens - token_objects = get_tokens(users, service) - if not token_objects: - return [], users - success_users, tokens = zip(*token_objects) - # send notifications + if tokens == []: + raise ValueError("No tokens to send notifications to.") + params = (tokens, title, body, category, is_dev, is_shadow) + if delay: - send_delayed_notifications(tokens, title, body, service, is_dev, is_shadow, delay) + send_delayed_notifications(*params, delay=delay) else: - send_immediate_notifications(tokens, title, body, service, is_dev, is_shadow) - - if not users: # if to all users, can't be any failed pennkeys - return success_users, [] - failed_users = list(set(users) - set(success_users)) - return success_users, failed_users - - -def get_tokens(users=None, service=None): - """Returns list of token objects (with username & token value) for specified users""" - - token_objs = NotificationToken.objects.select_related("user").filter( - kind=NotificationToken.KIND_IOS # NOTE: until Android implementation - ) - if users: - token_objs = token_objs.filter(user__username__in=users) - if service: - token_objs = token_objs.filter( - notificationsetting__service=service, notificationsetting__enabled=True - ) - return token_objs.exclude(token="").values_list("user__username", "token") + send_immediate_notifications(*params) @shared_task(name="notifications.send_immediate_notifications") @@ -103,21 +78,21 @@ def send_delayed_notifications(tokens, title, body, category, is_dev, is_shadow, ) -def get_auth_key_path(): - return os.environ.get( - "IOS_KEY_PATH", # for dev purposes - os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "ios_key.p8"), +def get_auth_key_path(is_dev): + return os.path.join( + os.path.dirname(os.path.dirname(os.path.abspath(__file__))), + f"apns-{'dev' if is_dev else 'prod'}.pem", ) def get_client(is_dev): """Creates and returns APNsClient based on iOS credentials""" - auth_key_path = get_auth_key_path() - auth_key_id = "2VX9TC37TB" - team_id = "VU59R57FGM" - token_credentials = TokenCredentials( - auth_key_path=auth_key_path, auth_key_id=auth_key_id, team_id=team_id - ) - client = APNsClient(credentials=token_credentials, use_sandbox=is_dev) + # auth_key_path = get_auth_key_path() + # auth_key_id = "2VX9TC37TB" + # team_id = "VU59R57FGM" + # token_credentials = TokenCredentials( + # auth_key_path=auth_key_path, auth_key_id=auth_key_id, team_id=team_id + # ) + client = APNsClient(credentials=get_auth_key_path(is_dev), use_sandbox=is_dev) return client diff --git a/backend/user/serializers.py b/backend/user/serializers.py index 94a17def..3e57c7d6 100644 --- a/backend/user/serializers.py +++ b/backend/user/serializers.py @@ -1,40 +1,7 @@ from django.contrib.auth import get_user_model from rest_framework import serializers -from user.models import NotificationSetting, NotificationToken, Profile - - -class NotificationTokenSerializer(serializers.ModelSerializer): - class Meta: - model = NotificationToken - fields = ("id", "kind", "token") - - def create(self, validated_data): - validated_data["user"] = self.context["request"].user - token_obj = NotificationToken.objects.filter(user=validated_data["user"]).first() - if token_obj: - raise serializers.ValidationError(detail={"detail": "Token already created."}) - return super().create(validated_data) - - -class NotificationSettingSerializer(serializers.ModelSerializer): - class Meta: - model = NotificationSetting - fields = ("id", "service", "enabled") - - def create(self, validated_data): - validated_data["token"] = NotificationToken.objects.get(user=self.context["request"].user) - setting = NotificationSetting.objects.filter( - token=validated_data["token"], service=validated_data["service"] - ).first() - if setting: - raise serializers.ValidationError(detail={"detail": "Setting already created."}) - return super().create(validated_data) - - def update(self, instance, validated_data): - if instance.service != validated_data["service"]: - raise serializers.ValidationError(detail={"detail": "Cannot change setting service."}) - return super().update(instance, validated_data) +from user.models import Profile class ProfileSerializer(serializers.ModelSerializer): diff --git a/backend/user/urls.py b/backend/user/urls.py index 5e06de40..3c069309 100644 --- a/backend/user/urls.py +++ b/backend/user/urls.py @@ -2,10 +2,12 @@ from rest_framework import routers from user.views import ( + AndroidNotificationTokenView, ClearCookiesView, + IOSNotificationTokenView, NotificationAlertView, - NotificationSettingView, - NotificationTokenView, + NotificationServiceSettingView, + NotificationServiceView, UserView, ) @@ -13,10 +15,19 @@ app_name = "user" router = routers.DefaultRouter() -router.register(r"notifications/tokens", NotificationTokenView, basename="notiftokens") -router.register(r"notifications/settings", NotificationSettingView, basename="notifsettings") +# router.register(r"notifications/settings", NotificationSettingView, basename="notifsettings") additional_urls = [ + path("notifications/tokens/ios//", IOSNotificationTokenView.as_view(), name="ios-token"), + path( + "notifications/tokens/android//", + AndroidNotificationTokenView.as_view(), + name="android-token", + ), + path( + "notifications/settings/", NotificationServiceSettingView.as_view(), name="notif-settings" + ), + path("notifications/services/", NotificationServiceView.as_view(), name="notif-services"), path("me/", UserView.as_view(), name="user"), path("notifications/alerts/", NotificationAlertView.as_view(), name="alert"), path("clear-cookies/", ClearCookiesView.as_view(), name="clear-cookies"), diff --git a/backend/user/views.py b/backend/user/views.py index 48ea7e46..6f559794 100644 --- a/backend/user/views.py +++ b/backend/user/views.py @@ -1,20 +1,17 @@ +from abc import ABC + from django.contrib.auth import get_user_model +from django.db import transaction from django.http import HttpResponseRedirect -from django.shortcuts import get_object_or_404 from identity.permissions import B2BPermission -from rest_framework import generics, viewsets -from rest_framework.decorators import action +from rest_framework import generics from rest_framework.permissions import IsAuthenticated from rest_framework.response import Response from rest_framework.views import APIView -from user.models import NotificationSetting, NotificationToken +from user.models import AndroidNotificationToken, IOSNotificationToken, NotificationService from user.notifications import send_push_notifications -from user.serializers import ( - NotificationSettingSerializer, - NotificationTokenSerializer, - UserSerializer, -) +from user.serializers import UserSerializer User = get_user_model() @@ -40,64 +37,79 @@ def get_object(self): return self.request.user -class NotificationTokenView(viewsets.ModelViewSet): - """ - get: - Return notification tokens of user. - """ - +class NotificationTokenView(APIView, ABC): permission_classes = [IsAuthenticated] - serializer_class = NotificationTokenSerializer + queryset = None - def get_queryset(self): - return NotificationToken.objects.filter(user=self.request.user) + def get_defaults(self): + raise NotImplementedError # pragma: no cover + def post(self, request, token): + _, created = self.queryset.update_or_create(token=token, defaults=self.get_defaults()) + if created: + return Response({"detail": "Token created."}, status=201) + return Response({"detail": "Token updated."}) -class NotificationSettingView(viewsets.ModelViewSet): - """ - get: - Return notification settings of user. + def delete(self, request, token): + self.queryset.filter(token=token).delete() + return Response({"detail": "Token deleted."}) - post: - Creates/updates new notification setting of user for a specific service. - check: - Checks if user wants notification for specified serice. - """ +class IOSNotificationTokenView(NotificationTokenView): + queryset = IOSNotificationToken.objects.all() - permission_classes = [B2BPermission("urn:pennlabs:*") | IsAuthenticated] - serializer_class = NotificationSettingSerializer - - def is_authorized(self, request): - return request.user and request.user.is_authenticated - - def get_queryset(self): - if self.is_authorized(self.request): - return NotificationSetting.objects.filter(token__user=self.request.user) - return NotificationSetting.objects.none() - - @action(detail=True, methods=["get"]) - def check(self, request, pk=None): - """ - Returns whether the user wants notification for specified service. - :param pk: service name - """ - - if pk not in dict(NotificationSetting.SERVICE_OPTIONS): - return Response({"detail": "Invalid Parameters."}, status=400) - - pennkey = request.GET.get("pennkey") - user = ( - request.user - if self.is_authorized(request) - else get_object_or_404(User, username=pennkey) + def get_defaults(self): + is_dev = self.request.data.get("is_dev", False) + return {"user": self.request.user, "is_dev": is_dev} + + +class AndroidNotificationTokenView(NotificationTokenView): + queryset = AndroidNotificationToken.objects.all() + + def get_defaults(self): + return {"user": self.request.user} + + +class NotificationServiceSettingView(APIView): + permission_classes = [IsAuthenticated] + + def get(self, request): + user = request.user + services = NotificationService.objects.all().prefetch_related("enabled_users") + return Response( + { + service.name: service.enabled_users.filter(id=user.id).exists() + for service in services + } ) - token = NotificationToken.objects.filter(user=user).first() - if not token: - return Response({"service": pk, "enabled": False}) - setting, _ = NotificationSetting.objects.get_or_create(token=token, service=pk) - return Response(NotificationSettingSerializer(setting).data) + def put(self, request): + user = request.user + settings = request.data + if not isinstance(settings, dict) or not all( + isinstance(value, bool) for value in settings.values() + ): + return Response({"detail": "Invalid request"}, status=400) + + try: + with transaction.atomic(): + user.notificationservice_set.add( + *[service for service, enabled in settings.items() if enabled] + ) + user.notificationservice_set.remove( + *[service for service, enabled in settings.items() if not enabled] + ) + except Exception as e: + return Response({"detail": str(e)}, status=400) + return Response({"detail": "Settings updated."}) + + +class NotificationServiceView(APIView): + permission_classes = [IsAuthenticated] + + # TODO: this is becoming a common pattern, consider abstracting + def get(self, request): + return Response(NotificationService.objects.all().values_list("name", flat=True)) class NotificationAlertView(APIView): @@ -109,11 +121,12 @@ class NotificationAlertView(APIView): permission_classes = [B2BPermission("urn:pennlabs:*") | IsAuthenticated] def post(self, request): - users = ( + usernames = ( [self.request.user.username] if request.user and request.user.is_authenticated else request.data.get("users", list()) ) + service = request.data.get("service") title = request.data.get("title") body = request.data.get("body") @@ -122,14 +135,29 @@ def post(self, request): if None in [service, title, body]: return Response({"detail": "Missing required parameters."}, status=400) - if service not in dict(NotificationSetting.SERVICE_OPTIONS): + + if not (service_obj := NotificationService.objects.filter(name=service).first()): return Response({"detail": "Invalid service."}, status=400) - success_users, failed_users = send_push_notifications( - users, service, title, body, delay, is_dev + users_with_service = service_obj.enabled_users.filter(username__in=usernames) + + tokens = list( + IOSNotificationToken.objects.filter( + user__in=users_with_service, is_dev=is_dev + ).values_list("token", flat=True) ) - return Response({"success_users": success_users, "failed_users": failed_users}) + if tokens: + send_push_notifications(tokens, service, title, body, delay, is_dev) + + users_with_service_usernames = users_with_service.values_list("username", flat=True) + users_not_reached_usernames = list(set(usernames) - set(users_with_service_usernames)) + return Response( + { + "success_users": users_with_service_usernames, + "failed_users": users_not_reached_usernames, + } + ) class ClearCookiesView(APIView): diff --git a/k8s/main.ts b/k8s/main.ts index f8b62811..55993bac 100644 --- a/k8s/main.ts +++ b/k8s/main.ts @@ -82,13 +82,13 @@ export class MyChart extends PennLabsChart { env: [{ name: "DJANGO_SETTINGS_MODULE", value: "pennmobile.settings.production" }] }); - new CronJob(this, 'send-gsr-reminders', { - schedule: "20,50 * * * *", - image: backendImage, - secret, - cmd: ["python", "manage.py", "send_gsr_reminders"], - env: [{ name: "DJANGO_SETTINGS_MODULE", value: "pennmobile.settings.production" }] - }); + // new CronJob(this, 'send-gsr-reminders', { + // schedule: "20,50 * * * *", + // image: backendImage, + // secret, + // cmd: ["python", "manage.py", "send_gsr_reminders"], + // env: [{ name: "DJANGO_SETTINGS_MODULE", value: "pennmobile.settings.production" }] + // }); new CronJob(this, 'get-fitness-snapshot', { schedule: cronTime.every(3).hours(), From d1044c2072f5514e4390ba5c93227b64cdb4ae35 Mon Sep 17 00:00:00 2001 From: Zachary Harpaz <61989294+zachHarpaz@users.noreply.github.com> Date: Sun, 17 Nov 2024 12:46:54 -0500 Subject: [PATCH 2/2] Created models for penn wrapped -- issue with pip files though (#309) * Created models * pip file fix * Fixed models.py and added to admin.py * models * Penn-Wrapped Backend Requests * fix github pipfile diff for no reason + others * Updated Models Finished Models for wrapped and serializer working * Fixes * Penn-Wrapped Prod Push Request Format: /wrapped/generate/semester=2020A --------- Co-authored-by: vcai122 --- backend/pennmobile/settings/base.py | 1 + backend/pennmobile/urls.py | 1 + backend/wrapped/__init__.py | 0 backend/wrapped/admin.py | 45 +++++ backend/wrapped/apps.py | 6 + backend/wrapped/migrations/0001_initial.py | 172 ++++++++++++++++++ ..._page_globalstatpagefield_page_and_more.py | 30 +++ backend/wrapped/migrations/__init__.py | 0 backend/wrapped/models.py | 107 +++++++++++ backend/wrapped/serializers.py | 69 +++++++ backend/wrapped/tests.py | 0 backend/wrapped/urls.py | 8 + backend/wrapped/views.py | 15 ++ 13 files changed, 454 insertions(+) create mode 100644 backend/wrapped/__init__.py create mode 100644 backend/wrapped/admin.py create mode 100644 backend/wrapped/apps.py create mode 100644 backend/wrapped/migrations/0001_initial.py create mode 100644 backend/wrapped/migrations/0002_rename_page_globalstatpagefield_page_and_more.py create mode 100644 backend/wrapped/migrations/__init__.py create mode 100644 backend/wrapped/models.py create mode 100644 backend/wrapped/serializers.py create mode 100644 backend/wrapped/tests.py create mode 100644 backend/wrapped/urls.py create mode 100644 backend/wrapped/views.py diff --git a/backend/pennmobile/settings/base.py b/backend/pennmobile/settings/base.py index cf285a89..c1bcd22a 100644 --- a/backend/pennmobile/settings/base.py +++ b/backend/pennmobile/settings/base.py @@ -46,6 +46,7 @@ "accounts.apps.AccountsConfig", "identity.apps.IdentityConfig", "analytics.apps.AnalyticsConfig", + "wrapped.apps.WrappedConfig", "django_filters", "debug_toolbar", "gsr_booking", diff --git a/backend/pennmobile/urls.py b/backend/pennmobile/urls.py index 5e94960e..f517a8af 100644 --- a/backend/pennmobile/urls.py +++ b/backend/pennmobile/urls.py @@ -27,6 +27,7 @@ path("dining/", include("dining.urls")), path("penndata/", include("penndata.urls")), path("sublet/", include("sublet.urls")), + path("wrapped/", include("wrapped.urls")), ] urlpatterns = [ diff --git a/backend/wrapped/__init__.py b/backend/wrapped/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/backend/wrapped/admin.py b/backend/wrapped/admin.py new file mode 100644 index 00000000..24a043ba --- /dev/null +++ b/backend/wrapped/admin.py @@ -0,0 +1,45 @@ +from django.contrib import admin + +from wrapped.models import ( + GlobalStat, + GlobalStatKey, + GlobalStatPageField, + IndividualStat, + IndividualStatKey, + IndividualStatPageField, + Page, + Semester, +) + + +class WrappedIndividualAdmin(admin.ModelAdmin): + search_fields = ["user__username__icontains", "key__key__icontains", "semester__icontains"] + list_display = ["user", "key", "value", "semester"] + + +class WrappedGlobalAdmin(admin.ModelAdmin): + + list_display = ["key", "value", "semester"] + search_fields = ["key__icontains"] + + +class IndividualStatPageFieldAdmin(admin.TabularInline): + model = IndividualStatPageField + extra = 1 + + +class GlobalStatPageFieldAdmin(admin.TabularInline): + model = GlobalStatPageField + extra = 1 + + +class PageAdmin(admin.ModelAdmin): + inlines = [IndividualStatPageFieldAdmin, GlobalStatPageFieldAdmin] + + +admin.site.register(IndividualStat, WrappedIndividualAdmin) +admin.site.register(GlobalStat, WrappedGlobalAdmin) +admin.site.register(IndividualStatKey) +admin.site.register(GlobalStatKey) +admin.site.register(Page, PageAdmin) +admin.site.register(Semester) diff --git a/backend/wrapped/apps.py b/backend/wrapped/apps.py new file mode 100644 index 00000000..7b3d895a --- /dev/null +++ b/backend/wrapped/apps.py @@ -0,0 +1,6 @@ +from django.apps import AppConfig + + +class WrappedConfig(AppConfig): + default_auto_field = "django.db.models.BigAutoField" + name = "wrapped" diff --git a/backend/wrapped/migrations/0001_initial.py b/backend/wrapped/migrations/0001_initial.py new file mode 100644 index 00000000..a69873cb --- /dev/null +++ b/backend/wrapped/migrations/0001_initial.py @@ -0,0 +1,172 @@ +# Generated by Django 5.0.2 on 2024-11-10 16:21 + +import datetime + +import django.db.models.deletion +from django.conf import settings +from django.db import migrations, models + + +class Migration(migrations.Migration): + + initial = True + + dependencies = [ + migrations.swappable_dependency(settings.AUTH_USER_MODEL), + ] + + operations = [ + migrations.CreateModel( + name="GlobalStatKey", + fields=[ + ("key", models.CharField(max_length=50, primary_key=True, serialize=False)), + ], + ), + migrations.CreateModel( + name="IndividualStatKey", + fields=[ + ("key", models.CharField(max_length=50, primary_key=True, serialize=False)), + ], + ), + migrations.CreateModel( + name="GlobalStatPageField", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, primary_key=True, serialize=False, verbose_name="ID" + ), + ), + ("text_field_name", models.CharField(max_length=50)), + ( + "global_stat_key", + models.ForeignKey( + default=None, + on_delete=django.db.models.deletion.CASCADE, + to="wrapped.globalstatkey", + ), + ), + ], + ), + migrations.CreateModel( + name="IndividualStatPageField", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, primary_key=True, serialize=False, verbose_name="ID" + ), + ), + ("text_field_name", models.CharField(max_length=50)), + ( + "individual_stat_key", + models.ForeignKey( + default=None, + on_delete=django.db.models.deletion.CASCADE, + to="wrapped.individualstatkey", + ), + ), + ], + ), + migrations.CreateModel( + name="Page", + fields=[ + ("name", models.CharField(max_length=50, primary_key=True, serialize=False)), + ("template_path", models.CharField(max_length=50)), + ("duration", models.DurationField(blank=True, default=datetime.timedelta(0))), + ( + "global_stats", + models.ManyToManyField( + blank=True, + through="wrapped.GlobalStatPageField", + to="wrapped.globalstatkey", + ), + ), + ( + "individual_stats", + models.ManyToManyField( + blank=True, + through="wrapped.IndividualStatPageField", + to="wrapped.individualstatkey", + ), + ), + ], + ), + migrations.AddField( + model_name="individualstatpagefield", + name="Page", + field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to="wrapped.page"), + ), + migrations.AddField( + model_name="globalstatpagefield", + name="Page", + field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to="wrapped.page"), + ), + migrations.CreateModel( + name="Semester", + fields=[ + ("semester", models.CharField(max_length=5, primary_key=True, serialize=False)), + ("pages", models.ManyToManyField(blank=True, to="wrapped.page")), + ], + ), + migrations.CreateModel( + name="IndividualStat", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, primary_key=True, serialize=False, verbose_name="ID" + ), + ), + ("value", models.CharField(max_length=50)), + ( + "user", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL + ), + ), + ( + "key", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="wrapped.individualstatkey" + ), + ), + ( + "semester", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="wrapped.semester" + ), + ), + ], + options={ + "unique_together": {("key", "semester", "user")}, + }, + ), + migrations.CreateModel( + name="GlobalStat", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, primary_key=True, serialize=False, verbose_name="ID" + ), + ), + ("value", models.CharField(max_length=50)), + ( + "key", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="wrapped.globalstatkey" + ), + ), + ( + "semester", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="wrapped.semester" + ), + ), + ], + options={ + "unique_together": {("key", "semester")}, + }, + ), + ] diff --git a/backend/wrapped/migrations/0002_rename_page_globalstatpagefield_page_and_more.py b/backend/wrapped/migrations/0002_rename_page_globalstatpagefield_page_and_more.py new file mode 100644 index 00000000..dcfe7098 --- /dev/null +++ b/backend/wrapped/migrations/0002_rename_page_globalstatpagefield_page_and_more.py @@ -0,0 +1,30 @@ +# Generated by Django 5.0.2 on 2024-11-10 17:40 + +import datetime + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("wrapped", "0001_initial"), + ] + + operations = [ + migrations.RenameField( + model_name="globalstatpagefield", + old_name="Page", + new_name="page", + ), + migrations.RenameField( + model_name="individualstatpagefield", + old_name="Page", + new_name="page", + ), + migrations.AlterField( + model_name="page", + name="duration", + field=models.DurationField(blank=True, default=datetime.timedelta(0), null=True), + ), + ] diff --git a/backend/wrapped/migrations/__init__.py b/backend/wrapped/migrations/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/backend/wrapped/models.py b/backend/wrapped/models.py new file mode 100644 index 00000000..04445427 --- /dev/null +++ b/backend/wrapped/models.py @@ -0,0 +1,107 @@ +from datetime import timedelta + +from django.contrib.auth import get_user_model +from django.db import models + + +User = get_user_model() + + +class StatKey(models.Model): + key = models.CharField(max_length=50, primary_key=True, null=False, blank=False) + + def __str__(self) -> str: + return self.key + + class Meta: + abstract = True + + +class IndividualStatKey(StatKey): + pass + + +class GlobalStatKey(StatKey): + pass + + +class Semester(models.Model): + semester = models.CharField(max_length=5, primary_key=True, null=False, blank=False) + pages = models.ManyToManyField("Page", blank=True) + + +class GlobalStat(models.Model): + + key = models.ForeignKey(GlobalStatKey, on_delete=models.CASCADE) + value = models.CharField(max_length=50, null=False, blank=False) + semester = models.ForeignKey(Semester, on_delete=models.CASCADE) + + class Meta: + unique_together = ("key", "semester") + + def __str__(self): + return f"Global -- {self.key}-{str(self.semester)} : {self.value}" + + +class IndividualStat(models.Model): + user = models.ForeignKey(User, on_delete=models.CASCADE) + key = models.ForeignKey(IndividualStatKey, on_delete=models.CASCADE) + + value = models.CharField(max_length=50, null=False, blank=False) + semester = models.ForeignKey(Semester, on_delete=models.CASCADE) + + class Meta: + unique_together = ("key", "semester", "user") + + def __str__(self) -> str: + return f"User: {self.user} -- {self.key}-{str(self.semester)} : {self.value}" + + +class Page(models.Model): + + name = models.CharField(max_length=50, primary_key=True, null=False, blank=False) + template_path = models.CharField(max_length=50, null=False, blank=False) + individual_stats = models.ManyToManyField( + IndividualStatKey, through="IndividualStatPageField", blank=True + ) + global_stats = models.ManyToManyField(GlobalStatKey, through="GlobalStatPageField", blank=True) + duration = models.DurationField(blank=True, null=True, default=timedelta(minutes=0)) + + def __str__(self): + return f"{self.name}" + + +class IndividualStatPageField(models.Model): + individual_stat_key = models.ForeignKey( + IndividualStatKey, null=False, blank=False, default=None, on_delete=models.CASCADE + ) + page = models.ForeignKey(Page, null=False, blank=False, on_delete=models.CASCADE) + text_field_name = models.CharField(max_length=50, null=False, blank=False) + + def get_value(self, user, semester): + return ( + IndividualStat.objects.filter( + user=user, key=self.individual_stat_key, semester=semester + ) + .first() + .value + ) + + def __str__(self): + return f"{self.page} -> {self.text_field_name} : {self.individual_stat_key}" + + +class GlobalStatPageField(models.Model): + global_stat_key = models.ForeignKey( + GlobalStatKey, null=False, blank=False, default=None, on_delete=models.CASCADE + ) + page = models.ForeignKey(Page, null=False, blank=False, on_delete=models.CASCADE) + text_field_name = models.CharField(max_length=50, null=False, blank=False) + + def get_value(self, _user, semester): + return ( + GlobalStat.objects.filter(key=self.global_stat_key.key, semester=semester).first().value + ) + + def __str__(self): + return f"{self.page} -> {self.text_field_name} : {self.global_stat_key}" diff --git a/backend/wrapped/serializers.py b/backend/wrapped/serializers.py new file mode 100644 index 00000000..23856e8d --- /dev/null +++ b/backend/wrapped/serializers.py @@ -0,0 +1,69 @@ +from rest_framework import serializers + +from wrapped.models import GlobalStatPageField, IndividualStatPageField, Page, Semester + + +class PageFieldSerializer(serializers.ModelSerializer): + stat_value = serializers.SerializerMethodField() + + class Meta: + abstract = True + fields = ["text_field_name", "stat_value"] + + def get_stat_value(self, obj): + user = self.context.get("user") + semester = self.context.get("semester") + return obj.get_value(user, semester) + + +class IndividualStatPageFieldSerializer(PageFieldSerializer): + class Meta(PageFieldSerializer.Meta): + model = IndividualStatPageField + + +class GlobalStatPageFieldSerializer(PageFieldSerializer): + class Meta(PageFieldSerializer.Meta): + model = GlobalStatPageField + + +class PageSerializer(serializers.ModelSerializer): + + combined_stats = serializers.SerializerMethodField() + + class Meta: + model = Page + fields = ["name", "template_path", "combined_stats", "duration"] + + def get_combined_stats(self, obj): + if not (semester := self.context.get("semester", obj)): + return {} + user = self.context.get("user") + individual_stat_fields = IndividualStatPageFieldSerializer( + obj.individualstatpagefield_set.all(), + context={"semester": semester, "user": user}, + many=True, + ).data + + global_stat_fields = GlobalStatPageFieldSerializer( + obj.globalstatpagefield_set.all(), + context={"semester": semester, "user": user}, + many=True, + ).data + + field_list = individual_stat_fields + global_stat_fields + + return {field["text_field_name"]: field["stat_value"] for field in field_list} + + +class SemesterSerializer(serializers.ModelSerializer): + pages = serializers.SerializerMethodField() + + class Meta: + model = Semester + fields = ["semester", "pages"] + + def get_pages(self, obj): + user = self.context.get("user") + return PageSerializer( + obj.pages.all(), many=True, context={"semester": obj, "user": user} + ).data diff --git a/backend/wrapped/tests.py b/backend/wrapped/tests.py new file mode 100644 index 00000000..e69de29b diff --git a/backend/wrapped/urls.py b/backend/wrapped/urls.py new file mode 100644 index 00000000..4f1e5850 --- /dev/null +++ b/backend/wrapped/urls.py @@ -0,0 +1,8 @@ +from django.urls import path + +from wrapped.views import SemesterView + + +urlpatterns = [ + path("semester//", SemesterView.as_view(), name="semester-detail"), +] diff --git a/backend/wrapped/views.py b/backend/wrapped/views.py new file mode 100644 index 00000000..defb0769 --- /dev/null +++ b/backend/wrapped/views.py @@ -0,0 +1,15 @@ +from rest_framework.permissions import IsAuthenticated +from rest_framework.response import Response +from rest_framework.views import APIView + +from wrapped.models import Semester +from wrapped.serializers import SemesterSerializer + + +class SemesterView(APIView): + permission_classes = [IsAuthenticated] + + def get(self, request, semester_id): + semester = Semester.objects.get(semester=semester_id) + serializer = SemesterSerializer(semester, context={"user": request.user}) + return Response(serializer.data)