from rest_framework import generics, status
from rest_framework.response import Response

from core.apps.notifications.serializers import notification as serializers
from core.apps.notifications.models import Notification


class RegisterExpoPushToken(generics.GenericAPIView):
    serializer_class = serializers.NotificationSerializer
    queryset = Notification.objects.all()

    def post(self, request):
        serializer = self.serializer_class(data=request.data)
        if serializer.is_valid():
            Notification.objects.update_or_create(
                user=request.user,
                token=serializer.validated_data['token']
            )
            return Response({"message": "Token saqlandi"}, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)