"""This module contains the api-v2 related serializers for the ``documents`` app"""

from rest_framework import serializers

from mtp.apps.trainings.apiv2.serializers import TrainingActivitySerializer

from ..models import Document


class DocumentSerializer(TrainingActivitySerializer, serializers.ModelSerializer):
    """Serializer for the ``Document`` model"""

    class Meta:
        model = Document
        fields = TrainingActivitySerializer.Meta.fields + (
            'trainee_document_file',
            'pdf_images_url_list',
            'needs_approval'
        )
