83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from rest_framework import permissions, status
|
|
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 ..serializers.ProfileSerializer import ProfileSerializer
|
|
|
|
|
|
class GetFriendsView(APIView):
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
authentication_classes = (SessionAuthentication,)
|
|
|
|
def get(self, request):
|
|
return Response(ProfileSerializer(request.user.profilemodel.get_friends(), many=True).data)
|
|
|
|
|
|
class EditFriendView(APIView):
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
authentication_classes = (SessionAuthentication,)
|
|
|
|
def get_object(self):
|
|
return self.request.user.profilemodel
|
|
|
|
def post(self, request, pk=None):
|
|
user_profile: ProfileModel = self.get_object()
|
|
friend_profile = get_object_or_404(ProfileModel, pk=pk)
|
|
|
|
if user_profile.pk == pk:
|
|
return Response(_('You can\'t be friend with yourself.'), status.HTTP_400_BAD_REQUEST)
|
|
|
|
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)
|
|
|
|
def delete(self, request, pk=None):
|
|
user_profile = self.get_object()
|
|
friend_profile = get_object_or_404(ProfileModel, pk=pk)
|
|
|
|
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)
|