mirror of
https://github.com/leaders-of-digital-9-task/backend.git
synced 2024-11-22 17:36:32 +03:00
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from django.contrib.auth import get_user_model
|
|
from rest_framework import generics, status
|
|
from rest_framework.decorators import action
|
|
from rest_framework.mixins import ListModelMixin, RetrieveModelMixin, UpdateModelMixin
|
|
from rest_framework.permissions import AllowAny
|
|
from rest_framework.response import Response
|
|
from rest_framework.viewsets import GenericViewSet
|
|
|
|
from .serializers import RegisterSerializer, UserSerializer
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class UserViewSet(RetrieveModelMixin, ListModelMixin, UpdateModelMixin, GenericViewSet):
|
|
serializer_class = UserSerializer
|
|
queryset = User.objects.all()
|
|
lookup_field = "username"
|
|
|
|
def get_queryset(self, *args, **kwargs):
|
|
return self.queryset.filter(id=self.request.user.id)
|
|
|
|
@action(detail=False)
|
|
def me(self, request):
|
|
serializer = UserSerializer(request.user, context={"request": request})
|
|
return Response(status=status.HTTP_200_OK, data=serializer.data)
|
|
|
|
|
|
class RegisterView(generics.CreateAPIView):
|
|
queryset = User.objects.all()
|
|
permission_classes = (AllowAny,)
|
|
serializer_class = RegisterSerializer
|