from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response

from userausfall.models import User
from userausfall.rest_api.serializers import UserSerializer


class UserViewSet(viewsets.ModelViewSet):
    class Meta:
        queryset = User.objects.all()

    @action(detail=False, methods=["PATCH"])
    def me(self, request):
        return Response(
            UserSerializer(
                instance=request.user, context={"request": request}
            ).data
        )