from app.entities import models
from rest_framework import viewsets, status
from rest_framework.response import Response
from django_filters import rest_framework as filters
from django_filters.rest_framework import DjangoFilterBackend
from rest_framework.filters import SearchFilter, OrderingFilter
from app.applicationlayer.cms.template.stakeholder.table_filters import StakeHolderFilter
from app.applicationlayer.cms.template.stakeholder import serializers
from app.applicationlayer.utils import (
    CustomPagination, status_message_response
)


class ChangeRequestTemplateStakeholdersViewset(viewsets.ModelViewSet):

    # queryset = models.ChangeRequestTemplateStakeHolders.objects.all()
    queryset = models.ChangeRequestTemplateStakeHolders.objects.select_related(
        'template_no'
    ).all()
    serializer_class = serializers.ChangeRequestTemplateStakeHoldersSerializer
    pagination_class = CustomPagination
    lookup_field = "code"
    filter_backends = (DjangoFilterBackend, SearchFilter, OrderingFilter)
    filterset_class = StakeHolderFilter
    ordering_fields = '__all__'
    search_fields = (
        'name', 'code',
        'department__company__name',
        'department__name',
        'email', 'contact_no'
    )

    def list(self, request, *args, **kwargs):

        queryset = self.filter_queryset(self.get_queryset())
        
        page = self.paginate_queryset(queryset)

        if page is not None:
            serializer = self.get_serializer(page, many=True)

            message = status_message_response(
                200,
                'success',
                'List of Templates Stakeholders found',
                serializer.data
            )

            return self.get_paginated_response(message)

        serializer = self.get_serializer(self.queryset, many=True)

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

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