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 = ('public', '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'