[Feat] add block friend

zakarm · May 5, 2024 · b8ea152 · b8ea152
1 parent b5fe4fd
commit b8ea152
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/app/back-end/dashboards/urls.py b/app/back-end/dashboards/urls.py
@@ -6,7 +6,8 @@
                     UserSearchView,
                     RemoveFriendshipView,
                     AcceptFriendshipView,
-                    AddFriendshipView
+                    AddFriendshipView,
+                    BlockFriendshipView
                    )
 
 urlpatterns = [
@@ -18,4 +19,5 @@
    path('friends-remove', RemoveFriendshipView.as_view(), name="friends-remove"),
    path('friends-accept', AcceptFriendshipView.as_view(), name="friends-accept"),
    path('friends-add', AddFriendshipView.as_view(), name="friends-add"),
+   path('friends-block', BlockFriendshipView.as_view(), name="friends-add"),
 ]
diff --git a/app/back-end/dashboards/views.py b/app/back-end/dashboards/views.py
@@ -135,3 +135,34 @@ def post(self, request):
             return Response({'success': 'Friendship Added'}, status=status.HTTP_200_OK)
         except Friendship.DoesNotExist:
             return Response({'error': 'Friendship does not exist'}, status=status.HTTP_400_BAD_REQUEST)
+
+class BlockFriendshipView(APIView):
+    authentication_classes = [JWTAuthentication]
+    permission_classes = [IsAuthenticated]
+
+    def block_friend(self, friendship, user_from):
+        if friendship.user_from == user_from:
+            block_flag = 'u_one_is_blocked_u_two'
+        else:
+            block_flag = 'u_two_is_blocked_u_one'
+
+        if getattr(friendship, block_flag):
+            return Response({'error': 'Friend already blocked'}, status=status.HTTP_400_BAD_REQUEST)
+        else:
+            setattr(friendship, block_flag, True)
+            friendship.save()
+            return Response({'success': 'Friend blocked'}, status=status.HTTP_200_OK)
+
+    def post(self, request):
+        user_from = request.user
+        username = request.data.get('username')
+        try:
+            user_accept = User.objects.get(username=username)
+        except User.DoesNotExist:
+            return Response({'error': 'User does not exist'}, status=status.HTTP_400_BAD_REQUEST)
+        try:
+            friendship = Friendship.objects.get(Q(user_from=user_from, user_to=user_accept)|
+                                                Q(user_from=user_accept, user_to=user_from))
+            return self.block_friend(friendship, user_from)
+        except Friendship.DoesNotExist:
+            return Response({'error': 'Friendship does not exist'}, status=status.HTTP_400_BAD_REQUEST)