from rest_framework import serializers
from translation import models
from utils.models import get_current_locale


class LanguageSerializer(serializers.ModelSerializer):
    """Serializer for model Language"""

    class Meta:
        model = models.Language
        fields = [
            'id',
            'title',
            'locale'
        ]


class SiteInterfaceDictionarySerializer(serializers.ModelSerializer):
    """Serializer for model SiteInterfaceDictionary."""

    text_trans = serializers.SerializerMethodField(read_only=True)
    text = serializers.JSONField(write_only=True)

    def get_text_trans(self, obj):
        return obj.text.get(get_current_locale(), '') if obj.text else ''

    class Meta:
        """Meta class."""
        model = models.SiteInterfaceDictionary
        fields = [
            'id',
            'page',
            'keywords',
            'text',
            'text_trans',
        ]