Add Models Path

pennlabs · Nov 19, 2024 · 0c65917 · 0c65917
1 parent 764c24b
commit 0c65917
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/user/models.py b/backend/user/models.py
@@ -48,4 +48,5 @@ 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)
-    NotificationService.objects.get(name="COURSES").enabled_users.add(instance)
+    if courses_service := NotificationService.objects.filter(name="COURSES").first():
+        courses_service.enabled_users.add(instance)