profiles: serializers representation to parse avatar url

profiles/viewsets/MyProfileViewSet.py

@@ -0,0 +1,39 @@
+from rest_framework import permissions
+from rest_framework import viewsets
+from rest_framework.response import Response
+from rest_framework.authentication import SessionAuthentication
+
+from ..serializers import ProfileSerializer
+from ..models import ProfileModel
+
+
+class MyProfileViewSet(viewsets.ModelViewSet):
+    permission_classes = (permissions.IsAuthenticated,)
+    authentication_classes = (SessionAuthentication,)
+    serializer_class = ProfileSerializer
+    queryset = ProfileModel.objects.all()
+
+    def get_object(self):
+        obj = self.get_queryset().get(pk=self.request.user.pk)
+        return obj
+
+    def perform_update(self, serializer: ProfileSerializer, pk=None):
+        serializer.is_valid(raise_exception=True)
+        avatar = serializer.validated_data.get('avatar')
+        profile: ProfileModel = self.get_object()
+
+        if (avatar is not None):
+            if (profile.avatar.name != "./profiles/static/avatars/default.avif"):
+                profile.avatar.storage.delete(profile.avatar.name)
+            serializer.save()
+
+    def delete_avatar(self, pk=None):
+        profile = self.get_object()
+        if (profile.avatar.name != './profiles/static/avatars/default.avif'):
+            profile.avatar.storage.delete(profile.avatar.name)
+        profile.avatar.name = './profiles/static/avatars/default.avif'
+        profile.save()
+        return Response(ProfileSerializer(profile).data)
+
+    def retrieve(self, pk=None):
+        return Response(self.serializer_class(self.get_object()).data)

profiles/viewsets/ProfileViewSet.py

@@ -0,0 +1,36 @@
+from django.utils.translation import gettext as _
+from django.contrib.auth.models import User
+
+from rest_framework import permissions, status
+from rest_framework import viewsets
+from rest_framework.response import Response
+
+from ..serializers import ProfileSerializer
+from ..models import ProfileModel
+
+
+class ProfileViewSet(viewsets.ModelViewSet):
+    queryset = ProfileModel.objects.all()
+    serializer_class = ProfileSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
+
+    def retrieve(self, request, username=None):
+        query = User.objects.filter(username=username)
+        if (not query):
+            return Response({"detail": _("Profile not found.")}, status.HTTP_404_NOT_FOUND)
+        query = self.get_queryset().filter(pk=query[0].pk)
+        if (not query):
+            return Response({"detail": _("Profile not found.")}, status.HTTP_404_NOT_FOUND)
+        instance = query[0]
+        return Response(self.serializer_class(instance).data)
+
+    def retrieve_id(self, request, pk=None):
+        query = self.get_queryset().filter(pk=pk)
+        if (not query):
+            return Response({"detail": _("Profile not found.")}, status.HTTP_404_NOT_FOUND)
+        instance = query[0]
+        return Response(self.serializer_class(instance).data)
+
+    def list(self, request):
+        serializer = ProfileSerializer(self.get_queryset(), many=True)
+        return Response(serializer.data)