from iconolab.models import AnnotationRevision, IconolabComment, AnnotationStats

from rest_framework import serializers

class AnnotationStatsSerializer(serializers.ModelSerializer):

    class Meta:

        model = AnnotationStats

        fields = ('submitted_revisions_count', 'awaiting_revisions_count', 'accepted_revisions_count', 'contributors_count', 'views_count', 'comments_count', 'tag_count')

class AnnotationRevisionSerializer(serializers.ModelSerializer):

    tags = serializers.SerializerMethodField('get_normalized_tags')

    annotation_guid = serializers.SerializerMethodField()

    stats = AnnotationStatsSerializer(source='annotation.stats')

    author = serializers.SerializerMethodField()

    def get_normalized_tags(self, obj):

        tags = []

        for tagging_info in obj.tagginginfo_set.all():

            tags.append({

                "tag_label": tagging_info.tag.label,

                "tag_link": tagging_info.tag.link,

                "accuracy": tagging_info.accuracy,

                "relevancy": tagging_info.relevancy,

            })

        return tags

    def get_annotation_guid(self, obj):

        return obj.annotation.annotation_guid

    def get_author(self, obj):

        return obj.author.username

    class Meta:

        model = AnnotationRevision

        fields = ('annotation_guid', 'title', 'description', 'fragment', 'tags', 'stats', 'created', 'author')

class IconolabCommentSerializer(serializers.ModelSerializer):

    allow_thread = serializers.BooleanField()

    class Meta:

        model = IconolabComment

        fields = '__all__'