profiles: friends reworked

profiles/models/FriendModel.py

@@ -1,7 +1,8 @@
-from django.contrib.auth.models import User
 from django.db import models
 
+from . import ProfileModel
+
 
 class FriendModel(models.Model):
-    friend1 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friend1')
-    friend2 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friend2')
+    friend1 = models.ForeignKey(ProfileModel.ProfileModel, on_delete=models.CASCADE, related_name='friend1')
+    friend2 = models.ForeignKey(ProfileModel.ProfileModel, on_delete=models.CASCADE, related_name='friend2')

profiles/models/ProfileModel.py

@@ -1,21 +1,22 @@
+from __future__ import annotations
 from os.path import splitext
 
 from django.contrib.auth.models import User
-from django.db import models
+from django.db.models import Q, Model, CASCADE, ForeignKey, ImageField
 from django.db.models.signals import post_save, pre_delete
 from django.dispatch import receiver
 
 from games.consumers import game_manager
-from ..models.FriendModel import FriendModel
+from . import FriendModel
 
 
 def upload_to(instance, filename: str):
     return f"./profiles/static/avatars/{instance.pk}{splitext(filename)[1]}"
 
 
-class ProfileModel(models.Model):
-    user = models.ForeignKey(User, on_delete=models.CASCADE)
-    avatar = models.ImageField(upload_to=upload_to, default="./profiles/static/avatars/default.avif")
+class ProfileModel(Model):
+    user = ForeignKey(User, on_delete=CASCADE)
+    avatar = ImageField(upload_to=upload_to, default="./profiles/static/avatars/default.avif")
 
     def get_game(self) -> int:
         for game in game_manager._game_list:
@@ -24,20 +25,23 @@ class ProfileModel(models.Model):
                     return game.game_id
         return None
 
-    def getFriends(self):
-        friends = []
+    def get_friends(self) -> list[ProfileModel]:
+        friends: list[ProfileModel] = []
 
-        for friendship in FriendModel.objects.filter(models.Q(friend1=self.user) | models.Q(friend2=self.user)):
-            friends.append(friendship.friend1 if friendship.friend1 != self.user else friendship.friend2)
+        for friendship in FriendModel.FriendModel.objects.filter(Q(friend1=self) | Q(friend2=self)):
+            friends.append(friendship.friend1 if friendship.friend1 != self else friendship.friend2)
 
         return friends
 
-    def isFriend(self, user):
-        return user in self.getFriends()
+    def is_friend(self, friend):
+        return friend in self.get_friends()
 
-    def deleteFriend(self, user):
-        for friendship in FriendModel.objects.filter(models.Q(friend1=self.user) | models.Q(friend2=self.user)):
-            friendship.delete()
+    def delete_friend(self, friend):
+        print(friend)
+        FriendModel.FriendModel.objects.get(
+            (Q(friend1=self) & Q(friend2=friend)) |
+            (Q(friend2=self) & Q(friend1=friend))
+        ).delete()
 
 
 @receiver(pre_delete, sender=ProfileModel)

profiles/tests/blocks.py

@@ -6,18 +6,6 @@ from rest_framework import status
 from profiles.serializers.ProfileSerializer import ProfileSerializer
 from profiles.models.ProfileModel import ProfileModel
 
-class ProfileTest(TestCase):
-    def setUp(self):
-        self.user: User = User.objects.create_user('bozo', password='password')
-        self.user.save()
-        self.expected_response = {'avatar': '/static/avatars/default.avif', 'user_id': 1, 'username': 'bozo'}
-
-        self.url = "/api/profiles/user/"
-
-    def test_profile_create_on_user_created(self):
-        response: HttpResponse = self.client.get(self.url + self.user.username)
-        self.assertDictEqual(self.expected_response, response.json())
-
 
 class BlockTest(TestCase):
     def setUp(self):

profiles/tests/profiles.py

@@ -0,0 +1,16 @@
+from django.test import TestCase
+from django.http import HttpResponse
+from django.contrib.auth.models import User
+
+
+class ProfileTest(TestCase):
+    def setUp(self):
+        self.user: User = User.objects.create_user('bozo', password='password')
+        self.user.save()
+        self.expected_response = {'avatar': '/static/avatars/default.avif', 'user_id': 1, 'username': 'bozo'}
+
+        self.url = "/api/profiles/user/"
+
+    def test_profile_create_on_user_created(self):
+        response: HttpResponse = self.client.get(self.url + self.user.username)
+        self.assertDictEqual(self.expected_response, response.json())

profiles/urls.py

@@ -3,6 +3,7 @@ from django.urls import path
 from .viewsets.ProfileViewSet import ProfileViewSet
 from .viewsets.MyProfileViewSet import MyProfileViewSet
 from .views.blocks import GetBlocksView, EditBlocksView
+from .views.friends import GetFriendsView, EditFriendView
 
 urlpatterns = [
     path("settings", MyProfileViewSet.as_view({'patch': 'partial_update', 'delete': 'delete_avatar'}), name="my_profile_page"),
@@ -10,7 +11,8 @@ urlpatterns = [
     path("", ProfileViewSet.as_view({'get': 'list'}), name="profiles_list"),
     path("block", GetBlocksView.as_view(), name="block_page"),
     path("block/<int:pk>", EditBlocksView.as_view(), name="block_page"),
-    # path("friend", views.FriendsView.as_view(), name="friend_page"),
+    path("friends", GetFriendsView.as_view(), name="friends_list_page"),
+    path("friends/<int:pk>", EditFriendView.as_view(), name="friends_edit_page"),
     path("user/<str:username>", ProfileViewSet.as_view({'get': 'retrieve'}), name="profile_page"),
     path("id/<int:pk>", ProfileViewSet.as_view({'get': 'retrieve_id'}), name="profile_page"),
 ]

profiles/views/friends.py

@@ -4,16 +4,49 @@ from rest_framework import permissions, status
 from rest_framework.authentication import SessionAuthentication
 
 from django.utils.translation import gettext as _
+from django.db.models import Q
+from django.shortcuts import get_object_or_404
 
 from ..models.FriendModel import FriendModel
+from ..models.ProfileModel import ProfileModel
+from ..serializers.ProfileSerializer import ProfileSerializer
 
 
-class FriendsView(APIView):
+class GetFriendsView(APIView):
     permission_classes = (permissions.IsAuthenticated,)
     authentication_classes = (SessionAuthentication,)
 
     def get(self, request):
-        friends = FriendModel.getFriends(request.user.pk)
-        if (friends):
-            return Response({"friends": friends, "user_id": request.user.pk}, status=status.HTTP_200_OK)
-        return Response({}, status=status.HTTP_204_NO_CONTENT)
+        query = ProfileModel.objects.filter(user=request.user)
+        if not query.exists():
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+        friends = query[0].get_friends()
+        return Response(ProfileSerializer(friends, many=True).data)
+
+
+class EditFriendView(APIView):
+    permission_classes = (permissions.IsAuthenticated,)
+    authentication_classes = (SessionAuthentication,)
+
+    def get_object(self):
+        return ProfileModel.objects.get(pk=self.request.user.pk)
+
+    def post(self, request, pk=None):
+        user_profile = self.get_object()
+        friend_profile = get_object_or_404(ProfileModel, pk=pk)
+
+        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)
+
+    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)
+
+        user_profile.delete_friend(friend_profile)
+        return Response(_('Friendship succssfully deleted.'))