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

clean: profile viewsets

Browse Source
This commit is contained in:
AdrienLSH 2024-04-07 14:27:07 +02:00
parent 6cdc8b2542
commit 51f8dfcaa3
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
profiles/viewsets.py
View File

@ -1,27 +1,26 @@
from rest_framework.parsers import MultiPartParser, FormParser
from rest_framework import permissions, status from rest_framework import permissions, status
from rest_framework import viewsets from rest_framework import viewsets
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.authentication import SessionAuthentication from rest_framework.authentication import SessionAuthentication
from django.http import HttpRequest from django.http import HttpRequest
from django.db.models import QuerySet
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.utils.translation import gettext as _
from .serializers import ProfileSerializer from .serializers import ProfileSerializer
from .models import ProfileModel from .models import ProfileModel
class ProfileViewSet(viewsets.ModelViewSet): class ProfileViewSet(viewsets.ModelViewSet):
queryset = ProfileModel.objects.all queryset = ProfileModel.objects.all()
serializer_class = ProfileSerializer serializer_class = ProfileSerializer
parser_classes = (MultiPartParser, FormParser)
permission_classes = (permissions.IsAuthenticatedOrReadOnly,) permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
def retrieve(self, request: HttpRequest, username=None): def retrieve(self, request: HttpRequest, username=None):
user = User.objects.filter(username=username) user = User.objects.filter(username=username)
if (not user): if (not user):
return Response({"detail": "Profile not found."}, status=status.HTTP_404_NOT_FOUND) return Response({"detail": _("Profile not found.")}, status=status.HTTP_404_NOT_FOUND)
instance = self.queryset().get(pk=user[0].pk) instance = self.get_queryset().get(pk=user[0].pk)
instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:] instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:]
return Response(self.serializer_class(instance).data, return Response(self.serializer_class(instance).data,
status=status.HTTP_200_OK) status=status.HTTP_200_OK)
@ -29,14 +28,14 @@ class ProfileViewSet(viewsets.ModelViewSet):
def retrieve_id(self, request: HttpRequest, pk=None): def retrieve_id(self, request: HttpRequest, pk=None):
user = User.objects.filter(pk=pk) user = User.objects.filter(pk=pk)
if (not user): if (not user):
return Response({"detail": "Profile not found."}, status=status.HTTP_404_NOT_FOUND) return Response({"detail": _("Profile not found.")}, status=status.HTTP_404_NOT_FOUND)
instance = self.queryset().get(pk=user[0].pk) instance = self.get_queryset().get(pk=user[0].pk)
instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:] instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:]
return Response(self.serializer_class(instance).data, return Response(self.serializer_class(instance).data,
status=status.HTTP_200_OK) status=status.HTTP_200_OK)
def list(self, request: HttpRequest): def list(self, request: HttpRequest):
serializer = ProfileSerializer(self.queryset(), many=True) serializer = ProfileSerializer(self.get_queryset(), many=True)
for profile in serializer.data: for profile in serializer.data:
profile["avatar"] = profile["avatar"][profile["avatar"].find("static") - 1:] profile["avatar"] = profile["avatar"][profile["avatar"].find("static") - 1:]
return Response(serializer.data) return Response(serializer.data)
@ -49,7 +48,7 @@ class MyProfileViewSet(viewsets.ModelViewSet):
queryset = ProfileModel.objects.all() queryset = ProfileModel.objects.all()
def get_object(self): def get_object(self):
obj = self.queryset.get(pk=self.request.user.pk) obj = self.get_queryset().get(pk=self.request.user.pk)
return obj return obj
def perform_update(self, serializer: ProfileSerializer, pk=None): def perform_update(self, serializer: ProfileSerializer, pk=None):