26 lines
845 B
Python
26 lines
845 B
Python
from rest_framework import generics, status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from .models import User
|
|
from .serializers import UserSerializer, UserUpdateSerializer, FCMTokenSerializer
|
|
|
|
|
|
class MeView(generics.RetrieveUpdateAPIView):
|
|
def get_serializer_class(self):
|
|
if self.request.method == "PATCH":
|
|
return UserUpdateSerializer
|
|
return UserSerializer
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
|
|
class FCMTokenView(APIView):
|
|
def put(self, request):
|
|
serializer = FCMTokenSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
request.user.fcm_token = serializer.validated_data["token"]
|
|
request.user.save(update_fields=["fcm_token"])
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|