starnakin/42_ft_transcendence
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

_

Browse Source
This commit is contained in:
Xamora
2024-04-18 20:56:21 +02:00
parent abb7797e48
commit 6b82f00a14
5 changed files with 13 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
profiles/views/friends.py
View File

@ -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, FriendRequestModel
from ..models import ProfileModel, FriendModel
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: ProfileModel = self.get_object()
user_profile = self.get_object()
friend_profile = get_object_or_404(ProfileModel, pk=pk)
if user_profile.pk == pk:
@ -35,16 +35,8 @@ class EditFriendView(APIView):
if user_profile.is_friend(friend_profile):
return Response(_('You are already friend with this user.'), status.HTTP_400_BAD_REQUEST)
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)
FriendModel(friend1=user_profile, friend2=friend_profile).save()
return Response(_('Friendship succssfully created.'), status.HTTP_201_CREATED)
def delete(self, request, pk=None):
user_profile = self.get_object()
@ -53,30 +45,5 @@ 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)