from rest_framework import viewsets, mixins, status, serializers
from rest_framework.response import Response
from rest_framework.permissions import IsAuthenticated
from rest_framework.decorators import action
from datetime import datetime

from .models import Edicto, Precio
from .serializer import EdictoSerializer, PrecioSerializer
from .utils import contador
# from .precio import contador


class EdictoViewSet(mixins.CreateModelMixin,
                    mixins.RetrieveModelMixin,
                    mixins.UpdateModelMixin,
                    mixins.DestroyModelMixin,
                    mixins.ListModelMixin,
                    viewsets.GenericViewSet):

    queryset = Edicto.objects.all().order_by('id')
    serializer_class = EdictoSerializer
    permission_classes = [IsAuthenticated, ]

    def perform_create(self, serializer):
        edicto = self.request.data.get('cuerpo_edicto')
        cantidad_palabras = contador(edicto)
        serializer.save(cantidad_palabras=cantidad_palabras, usuario=self.request.user)

    def custom_get_method(self, request):

        instance = self.get_object()
    @action(detail=True, methods=['put', 'patch'])
    def custom_update(self, request):
        edicto = request.data.get('cuerpo_edicto')
        cantidad_palabras = contador(edicto)

        instance = self.get_object()
        serializer = self.get_serializer(instance, data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.save(cantidad_palabras=cantidad_palabras, fecha_actualizacion=datetime.now())

        return Response(serializer.data, status=status.HTTP_200_OK






class PrecioViewSet(mixins.CreateModelMixin,
                    mixins.RetrieveModelMixin,
                    mixins.ListModelMixin,
                    viewsets.GenericViewSet):

    queryset = Precio.objects.all().order_by('id')
    serializer_class = PrecioSerializer
    permission_classes = [IsAuthenticated, ]

    def create(self, request, *args, **kwargs):
        user = request.user

        if not user.is_staff:
            raise serializers.ValidationError("El usuario no es parte del personal designado.")

        return super().create(request, *args, **kwargs)

    def retrieve(self, request, *args, **kwargs):
        instance = self.get_object()
        serializer = self.get_serializer(instance)
        data = serializer.data
        data['id'] = instance.id
        return Response(data)