from rest_framework import serializers
from library.models import Article


class ArticleSerializer(serializers.ModelSerializer):
    tags = serializers.SerializerMethodField()
    section = serializers.SerializerMethodField()

    class Meta:
        model = Article
        exclude = ('css', 'js', 'page', 'head_title', 'public',
                   'head_description', 'head_image', 'id')

    @staticmethod
    def get_tags(self):
        return [i.name for i in self.tags.all()]

    @staticmethod
    def get_section(self):
        return self.section.name if self.section else 'undefined'