from drf_spectacular.utils import (
    extend_schema,
    extend_schema_view,
    OpenApiResponse,
    PolymorphicProxySerializer,
)

from itim.serializers.problem import (
    ProblemAddTicketModelSerializer,
    ProblemChangeTicketModelSerializer,
    ProblemImportTicketModelSerializer,
    ProblemTriageTicketModelSerializer,
    ProblemTicketViewSerializer,
)

from core.viewsets.ticket import TicketViewSet



@extend_schema_view(
    create=extend_schema(
        summary = 'Create a Problem Ticket',
        description='',
        request = PolymorphicProxySerializer(
            component_name = 'ProblemTicket',
            serializers=[
                ProblemImportTicketModelSerializer,
                ProblemAddTicketModelSerializer,
                ProblemChangeTicketModelSerializer,
                ProblemTriageTicketModelSerializer,
            ],
            resource_type_field_name=None,
            many = False
        ),
        responses = {
            201: OpenApiResponse(description='Created', response=ProblemTicketViewSerializer),
            403: OpenApiResponse(description='User is missing add permissions'),
        }
    ),
    destroy = extend_schema(
        summary = 'Delete a Problem Ticket',
        description = '',
        responses = {
            204: OpenApiResponse(description=''),
            403: OpenApiResponse(description='User is missing delete permissions'),
        }
    ),
    list = extend_schema(
        summary = 'Fetch all Problem Tickets',
        description='',
        responses = {
            200: OpenApiResponse(description='', response=ProblemTicketViewSerializer),
            403: OpenApiResponse(description='User is missing view permissions'),
        }
    ),
    retrieve = extend_schema(
        summary = 'Fetch a Problem Ticket',
        description='',
        responses = {
            200: OpenApiResponse(description='', response=ProblemTicketViewSerializer),
            403: OpenApiResponse(description='User is missing view permissions'),
        }
    ),
    update = extend_schema(exclude = True),
    partial_update = extend_schema(
        summary = 'Update a Problem Ticket',
        description = '',
        responses = {
            200: OpenApiResponse(description='', response=ProblemTicketViewSerializer),
            403: OpenApiResponse(description='User is missing change permissions'),
        }
    ),
)
class ViewSet(TicketViewSet):
    """Problem Ticket

    This class exists only for the purpose of swagger for documentation.

    Args:
        TicketViewSet (class): Base Ticket ViewSet.
    """

    _ticket_type: str = 'Problem'

    view_description = 'Problem Tickets'