add views for unsubscribing

Author: VeryBigSad

projects/urls.py

@@ -13,6 +13,7 @@
     ProjectRecommendedUsers,
     SetLikeOnProject,
     ProjectSubscribe,
+    ProjectUnsubscribe,
     ProjectSubscribers,
 )
 
@@ -23,6 +24,7 @@
     path("<int:pk>/like/", SetLikeOnProject.as_view()),
     path("<int:project_pk>/news/", NewsList.as_view()),
     path("<int:project_pk>/subscribe/", ProjectSubscribe.as_view()),
+    path("<int:project_pk>/unsubscribe/", ProjectUnsubscribe.as_view()),
     path("<int:project_pk>/subscribers/", ProjectSubscribers.as_view()),
     path("<int:project_pk>/news/<int:pk>/", NewsDetail.as_view()),
     path("<int:project_pk>/news/<int:pk>/set_viewed/", NewsDetailSetViewed.as_view()),

projects/views.py

@@ -420,11 +420,44 @@ def post(self, request, project_pk):
             project = Project.objects.get(pk=project_pk)
         except Project.DoesNotExist:
             raise NotFound
-
-        project.subscribers.add(request.user)
+        try:
+            project.subscribers.add(request.user)
+        except Exception:
+            return Response(
+                {
+                    "detail": f"User {request.user.id} is not part of project {project.pk}."
+                },
+                status=status.HTTP_400_BAD_REQUEST,
+            )
 
         logger.info(f"User {request.user.id} subscribed to project {project_pk}")
 
         return Response(
             {"detail": "Subscriber was successfully added"}, status=status.HTTP_200_OK
         )
+
+
+class ProjectUnsubscribe(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def post(self, request, project_pk):
+        try:
+            project = Project.objects.get(pk=project_pk)
+        except Project.DoesNotExist:
+            raise NotFound
+        try:
+            project.subscribers.remove(request.user)
+            # todo: add more specific error here
+        except Exception:
+            return Response(
+                {
+                    "detail": f"User {request.user.id} is not part of project {project.pk}."
+                },
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+        logger.info(f"User {request.user.id} unsubscribed to project {project_pk}")
+
+        return Response(
+            {"detail": "Subscriber was successfully removed"}, status=status.HTTP_200_OK
+        )