from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from drf_yasg.utils import swagger_auto_schema
from django.shortcuts import get_object_or_404

from producers_sub_categories.models.producers_sub_categories import ProducersSubCategories
from producers_sub_categories.serializers.producers_sub_categories_serializer import ProducersSubCategoriesSerializer
from common.pagination.custom_pagination import CustomPageNumberPagination
from common.exceptions.exceptions import custom_exception_handler

from rest_framework.parsers import MultiPartParser, FormParser
from rest_framework.permissions import AllowAny


class ProducersSubCategoriesListCreateAPIView(APIView):
    parser_classes = [MultiPartParser, FormParser]
    permission_classes = [AllowAny]

    @swagger_auto_schema(request_body=ProducersSubCategoriesSerializer)
    def post(self, request):
        serializer = ProducersSubCategoriesSerializer(data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)
        serializer.save()
        return Response({"Success": "Producers sub Category created successfully"}, status=status.HTTP_201_CREATED)
    
    def get(self, request):
        categories = ProducersSubCategories.objects.filter(is_active=True).select_related('category')
        category_by_name = self.request.query_params.get('category')
        if category_by_name:
            categories = categories.filter(category__id__iexact=category_by_name)
        paginator = CustomPageNumberPagination()
        result_page = paginator.paginate_queryset(categories, request)
        serializer = ProducersSubCategoriesSerializer(result_page, many=True, context={'request': request})
        return paginator.get_paginated_response(serializer.data)
    

class ProducersSubCategoriesDetailAPIView(APIView):
    parser_classes = [MultiPartParser, FormParser]
    permission_classes = [AllowAny]

    @swagger_auto_schema(request_body=ProducersSubCategoriesSerializer)
    def put(self, request, pk):
        category = get_object_or_404(ProducersSubCategories, pk=pk)
        serializer = ProducersSubCategoriesSerializer(category, data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)
        serializer.save()
        return Response({"Success": "Producers sub Category updated successfully"}, status=status.HTTP_200_OK)
    
    def get(self, request, pk):
        category = get_object_or_404(ProducersSubCategories.objects.select_related('category'), pk=pk)
        serializer = ProducersSubCategoriesSerializer(category, context={'request': request})
        return Response(serializer.data)

    def delete(self, request, pk):
        category = get_object_or_404(ProducersSubCategories, pk=pk)
        category.soft_delete()
        return Response({"success": "Producers sub Category deleted successfully"}, status=status.HTTP_200_OK)