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

profiles: use of OneToOneField to access profile through user

Browse Source
This commit is contained in:
AdrienLSH
2024-04-18 09:29:36 +02:00
parent 96b0d493e5
commit 5a2da91d6e
5 changed files with 34 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
profiles/views/friends.py
View File

@ -4,7 +4,6 @@ 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 import ProfileModel, FriendModel
@ -16,11 +15,7 @@ class GetFriendsView(APIView):
authentication_classes = (SessionAuthentication,)
def get(self, request):
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)
return Response(ProfileSerializer(request.user.profilemodel.get_friends(), many=True).data)
class EditFriendView(APIView):
@ -28,7 +23,7 @@ class EditFriendView(APIView):
authentication_classes = (SessionAuthentication,)
def get_object(self):
return ProfileModel.objects.get(pk=self.request.user.pk)
return self.request.user.profilemodel
def post(self, request, pk=None):
user_profile = self.get_object()