from django.http import Http404

from rest_framework import generics
from rest_framework.response import Response
from rest_framework.views import APIView 
from rest_framework import status

from app_venin_web_new1.serializers import UnidadesSerializer, UnidadesSerializerP
from app_venin_web_new1.models import Unidades



#Unidad
class UnidadesListView(generics.ListCreateAPIView):
    queryset = Unidades.objects.all()
    serializer_class = UnidadesSerializer
    name = "Unidades List View"

    def post(self, request, format=None):
        serializer = UnidadesSerializerP(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response({"message":"Unidad creada exitosamente.",
                             "data":serializer.data,
                             "status":"success"}, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

class UnidadesDetails(APIView):

    def get_object(self, pk):
        try:
            return Unidades.objects.get(pk=pk)
        except Unidades.DoesNotExist:
            raise Http404
    def get(self, request, pk, format=None):
        query = self.get_object(pk)
        serializer = UnidadesSerializer(query)
        return Response(serializer.data)
    def put(self, request, pk, format=None):
        query = self.get_object(pk)
        serializer = UnidadesSerializer(query, data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response({"message":"Datos actualizados exitosamente.",
                             "data":serializer.data,
                             "status":"success"})
        else:
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
    def delete(self, request, pk, format=None):
        query = self.get_object(pk)
        query.delete()
        return Response({"message":"Elemento eliminado exitosamente.",
                        "status":"success"},status=status.HTTP_200_OK) 