"""Search indexes serializers."""
from rest_framework import serializers
from django_elasticsearch_dsl_drf.serializers import DocumentSerializer
from search_indexes.documents.news import NewsDocument
from utils.models import get_current_language


class NewsDocumentSerializer(DocumentSerializer):
    """News document serialzier."""

    title_translated = serializers.SerializerMethodField(method_name='get_title_translated',
                                                         allow_null=True)

    class Meta:
        """Meta class."""

        document = NewsDocument
        fields = (
            'id',
            'title',
            'subtitle',
            'description',
            'web_url',
            'title_translated',
        )

    def get_title_translated(self, instance):
        return instance.title.get(get_current_language())