dockered

django/profiles/views/blocks.py

@@ -0,0 +1,57 @@
+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 rest_framework.request import Request
+
+from django.utils.translation import gettext as _
+from django.shortcuts import get_object_or_404
+
+from ..models import BlockModel, ProfileModel
+from ..serializers import ProfileSerializer
+
+
+class GetBlocksView(APIView):
+    permission_classes = (permissions.IsAuthenticated,)
+    authentication_classes = (SessionAuthentication,)
+
+    def get(self, request: Request):
+        blocks = BlockModel.objects.filter(blocker=request.user.profilemodel)
+        bloked_profiles = [block.blocked for block in blocks]
+
+        return Response(ProfileSerializer(bloked_profiles, many=True).data)
+
+
+class EditBlocksView(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 = self.get_object()
+        blocked_profile = get_object_or_404(ProfileModel, pk=pk)
+
+        if user_profile.pk == pk:
+            return Response(_('You can\'t block yourself.'), status.HTTP_400_BAD_REQUEST)
+
+        if BlockModel.objects.filter(blocker=user_profile, blocked=blocked_profile):
+            return Response(_('You already blocked this user.'), status.HTTP_409_CONFLICT)
+
+        BlockModel(blocker=user_profile, blocked=blocked_profile).save()
+        return Response(_('User successfully blocked.'), status.HTTP_201_CREATED)
+
+    def delete(self, request, pk=None):
+        user_profile = self.get_object()
+        blocked_profile = get_object_or_404(ProfileModel, pk=pk)
+
+        if user_profile.pk == pk:
+            return Response(_('You can\'t unblock yourself.'), status.HTTP_400_BAD_REQUEST)
+
+        block_record = BlockModel.objects.filter(blocker=user_profile, blocked=blocked_profile).first()
+        if not block_record:
+            return Response(_('This user is not blocked.'), status.HTTP_400_BAD_REQUEST)
+
+        block_record.delete()
+        return Response(_('User successfully unblocked.'), status.HTTP_200_OK)

django/profiles/views/friends.py

@@ -0,0 +1,87 @@
+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 import ProfileSerializer
+from notice.consumers import notice_manager
+
+
+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()
+            notice_manager.notify_new_friend(friend_profile.user, user_profile)
+            return Response(_('Friendship successfully created.'), status.HTTP_201_CREATED)
+
+        FriendRequestModel(author=user_profile, target=friend_profile).save()
+        notice_manager.notify_friend_request(friend_profile.user, user_profile)
+        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)
+
+        outgoing_request = user_profile.get_outgoing_friend_request_to(friend_profile)
+        if outgoing_request:
+            outgoing_request.delete()
+            notice_manager.notify_friend_request_canceled(friend_profile.user, user_profile)
+            return Response(_('Friend request cancelled.'))
+
+        if not user_profile.is_friend(friend_profile):
+            return Response(_('You are not friend with this user.'), status.HTTP_400_BAD_REQUEST)
+
+        user_profile.delete_friend(friend_profile)
+        notice_manager.notify_friend_removed(friend_profile.user, user_profile)
+        return Response(_('Friendship successfully deleted.'), status.HTTP_201_CREATED)
+
+
+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)