clean: profile viewsets
This commit is contained in:
parent
6cdc8b2542
commit
51f8dfcaa3
@ -1,27 +1,26 @@
|
||||
from rest_framework.parsers import MultiPartParser, FormParser
|
||||
from rest_framework import permissions, status
|
||||
from rest_framework import viewsets
|
||||
from rest_framework.response import Response
|
||||
from rest_framework.authentication import SessionAuthentication
|
||||
|
||||
from django.http import HttpRequest
|
||||
from django.db.models import QuerySet
|
||||
from django.contrib.auth.models import User
|
||||
from django.utils.translation import gettext as _
|
||||
|
||||
from .serializers import ProfileSerializer
|
||||
from .models import ProfileModel
|
||||
|
||||
|
||||
class ProfileViewSet(viewsets.ModelViewSet):
|
||||
queryset = ProfileModel.objects.all
|
||||
queryset = ProfileModel.objects.all()
|
||||
serializer_class = ProfileSerializer
|
||||
parser_classes = (MultiPartParser, FormParser)
|
||||
permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
|
||||
|
||||
def retrieve(self, request: HttpRequest, username=None):
|
||||
user = User.objects.filter(username=username)
|
||||
if (not user):
|
||||
return Response({"detail": "Profile not found."}, status=status.HTTP_404_NOT_FOUND)
|
||||
instance = self.queryset().get(pk=user[0].pk)
|
||||
return Response({"detail": _("Profile not found.")}, status=status.HTTP_404_NOT_FOUND)
|
||||
instance = self.get_queryset().get(pk=user[0].pk)
|
||||
instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:]
|
||||
return Response(self.serializer_class(instance).data,
|
||||
status=status.HTTP_200_OK)
|
||||
@ -29,14 +28,14 @@ class ProfileViewSet(viewsets.ModelViewSet):
|
||||
def retrieve_id(self, request: HttpRequest, pk=None):
|
||||
user = User.objects.filter(pk=pk)
|
||||
if (not user):
|
||||
return Response({"detail": "Profile not found."}, status=status.HTTP_404_NOT_FOUND)
|
||||
instance = self.queryset().get(pk=user[0].pk)
|
||||
return Response({"detail": _("Profile not found.")}, status=status.HTTP_404_NOT_FOUND)
|
||||
instance = self.get_queryset().get(pk=user[0].pk)
|
||||
instance.avatar.name = instance.avatar.name[instance.avatar.name.find("static") - 1:]
|
||||
return Response(self.serializer_class(instance).data,
|
||||
status=status.HTTP_200_OK)
|
||||
|
||||
def list(self, request: HttpRequest):
|
||||
serializer = ProfileSerializer(self.queryset(), many=True)
|
||||
serializer = ProfileSerializer(self.get_queryset(), many=True)
|
||||
for profile in serializer.data:
|
||||
profile["avatar"] = profile["avatar"][profile["avatar"].find("static") - 1:]
|
||||
return Response(serializer.data)
|
||||
@ -49,7 +48,7 @@ class MyProfileViewSet(viewsets.ModelViewSet):
|
||||
queryset = ProfileModel.objects.all()
|
||||
|
||||
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
|
||||
|
||||
def perform_update(self, serializer: ProfileSerializer, pk=None):
|
||||
|
Loading…
Reference in New Issue
Block a user