profiles: friend requests

profiles/views/friends.py

@@ -6,7 +6,7 @@ from rest_framework.authentication import SessionAuthentication
 from django.utils.translation import gettext as _
 from django.shortcuts import get_object_or_404
 
-from ..models import ProfileModel, FriendModel
+from ..models import ProfileModel, FriendRequestModel
 from ..serializers.ProfileSerializer import ProfileSerializer
 
 
@@ -26,7 +26,7 @@ class EditFriendView(APIView):
         return self.request.user.profilemodel
 
     def post(self, request, pk=None):
-        user_profile = self.get_object()
+        user_profile: ProfileModel = self.get_object()
         friend_profile = get_object_or_404(ProfileModel, pk=pk)
 
         if user_profile.pk == pk:
@@ -35,8 +35,16 @@ class EditFriendView(APIView):
         if user_profile.is_friend(friend_profile):
             return Response(_('You are already friend with this user.'), status.HTTP_400_BAD_REQUEST)
 
-        FriendModel(friend1=user_profile, friend2=friend_profile).save()
-        return Response(_('Friendship succssfully created.'), status.HTTP_201_CREATED)
+        if user_profile.is_friend_requesting(friend_profile):
+            return Response(_('You already sent a request to this user.'), status.HTTP_400_BAD_REQUEST)
+
+        incoming_request = user_profile.get_received_friend_request_from(friend_profile)
+        if incoming_request:
+            incoming_request.accept()
+            return Response(_('Friendship succssfully created.'), status.HTTP_201_CREATED)
+
+        FriendRequestModel(author=user_profile, target=friend_profile).save()
+        return Response(_('Friend request sent.'), status.HTTP_200_OK)
 
     def delete(self, request, pk=None):
         user_profile = self.get_object()
@@ -45,5 +53,30 @@ class EditFriendView(APIView):
         if not user_profile.is_friend(friend_profile):
             return Response(_('You are not friend with this user.'), status.HTTP_400_BAD_REQUEST)
 
+        outgoing_request = user_profile.get_outgoing_friend_request_to(friend_profile)
+        if outgoing_request:
+            outgoing_request.delete()
+            return Response(_('Friend request cancelled.'))
+
         user_profile.delete_friend(friend_profile)
         return Response(_('Friendship succssfully deleted.'))
+
+
+class GetIncomingFriendRequestView(APIView):
+    permission_classes = (permissions.IsAuthenticated,)
+    authentication_classes = (SessionAuthentication,)
+
+    def get(self, request):
+        requests = request.user.profilemodel.get_incoming_friend_requests()
+        profiles = [request.author for request in requests]
+        return Response(ProfileSerializer(profiles, many=True).data)
+
+
+class GetOutgoingFriendRequestView(APIView):
+    permission_classes = (permissions.IsAuthenticated,)
+    authentication_classes = (SessionAuthentication,)
+
+    def get(self, request):
+        requests = request.user.profilemodel.get_outgoing_friend_requests()
+        profiles = [request.target for request in requests]
+        return Response(ProfileSerializer(profiles, many=True).data)