from rest_framework import serializers

from vbv_lernwelt.duedate.models import DueDate


class DueDateSerializer(serializers.ModelSerializer):
    circle = serializers.SerializerMethodField()

    class Meta:
        model = DueDate
        fields = [
            "start",
            "end",
            "manual_override_fields",
            "title",
            "assignment_type_translation_key",
            "date_type_translation_key",
            "subtitle",
            "url",
            "course_session",
            "page",
            "circle",
        ]

    def get_circle(self, obj):
        circle = obj.get_circle()
        if circle:
            return {
                "id": circle.id,
                "title": circle.title,
                "translation_key": circle.translation_key,
            }
        return None