from drf_spectacular.utils import extend_schema, extend_schema_view, OpenApiResponse

from core.serializers.ticket_comment_category_notes import (
    TicketCommentCategoryNotes,
    TicketCommentCategoryNoteModelSerializer,
    TicketCommentCategoryNoteViewSerializer,
)

from core.viewsets.model_notes import ModelNoteViewSet



@extend_schema_view(
    create=extend_schema(
        summary = 'Add a note to a Ticket Comment Category',
        description = '',
        responses = {
            201: OpenApiResponse(description='created', response=TicketCommentCategoryNoteViewSerializer),
            400: OpenApiResponse(description='Validation failed.'),
            403: OpenApiResponse(description='User is missing create permissions'),
        }
    ),
    destroy = extend_schema(
        summary = 'Delete a Ticket  Comment Category note',
        description = ''
    ),
    list = extend_schema(
        summary = 'Fetch all Ticket Comment Category notes',
        description='',
    ),
    retrieve = extend_schema(
        summary = 'Fetch a single Ticket Comment Category note',
        description='',
    ),
    update = extend_schema(exclude = True),
    partial_update = extend_schema(
        summary = 'Update a Ticket Comment Category note',
        description = ''
    ),
)
class ViewSet(ModelNoteViewSet):

    model = TicketCommentCategoryNotes


    def get_serializer_class(self):

        if (
            self.action == 'list'
            or self.action == 'retrieve'
        ):

            self.serializer_class = TicketCommentCategoryNoteViewSerializer


        else:
            
            self.serializer_class = TicketCommentCategoryNoteModelSerializer

        return self.serializer_class
