from rest_framework import serializers

from vbv_lernwelt.course_session.models import CourseSessionAttendanceCourse


class CourseSessionAttendanceCourseSerializer(serializers.ModelSerializer):
    start = serializers.SerializerMethodField()
    end = serializers.SerializerMethodField()

    class Meta:
        model = CourseSessionAttendanceCourse
        fields = [
            "id",
            "course_session",
            "learning_content",
            "due_date",
            "location",
            "trainer",
            "start",
            "end",
        ]

    def get_start(self, obj):
        return obj.due_date.start

    def get_end(self, obj):
        return obj.due_date.end