svthalia/concrexit

from rest_framework import serializers

from events.models import EventRegistration
from members.api.v2.serializers.member import MemberSerializer
from members.models import Member
from payments.api.v2.serializers import PaymentSerializer
from thaliawebsite.api.v2.serializers.cleaned_model_serializer import (
    CleanedModelSerializer,
)


class EventRegistrationAdminSerializer(CleanedModelSerializer):
    """Serializer for event registrations."""

    class Meta:
        model = EventRegistration
        fields = (
            "pk",
            "present",
            "queue_position",
            "date",
            "date_cancelled",
            "payment",
            "member",
            "name",
            "special_price",
            "remarks",
            "phone_number",
            "email",
        )
        read_only_fields = ("payment",)
        optional_fields = ["payment", "member", "name", "special_price", "remarks"]

    payment = PaymentSerializer(required=False)

    def create(self, validated_data):
        event = self.context["event"]
        validated_data.update({"event": event})
        return super().create(validated_data)

    def to_internal_value(self, data):
        self.fields["member"] = serializers.PrimaryKeyRelatedField(
            queryset=Member.objects.all(), required=False
        )
        return super().to_internal_value(data)

    def to_representation(self, instance):
        self.fields["member"] = MemberSerializer(
            admin=True, detailed=False, read_only=True
        )
        return super().to_representation(instance)