svthalia/concrexit

from typing import Any

from django.db.models import Model
from django.utils.translation import gettext_lazy as _

from rest_framework import serializers
from rest_framework.exceptions import ValidationError

from payments.api.v1.fields import PaymentTypeField
from payments.models import Payment
from pizzas.models import FoodEvent, FoodOrder, Product
from pizzas.services import can_change_order
from thaliawebsite.api.v1.cleaned_model_serializer import CleanedModelSerializer


class PizzaSerializer(CleanedModelSerializer):
    class Meta:
        model = Product
        fields = ("pk", "name", "description", "price", "available")


class PizzaEventSerializer(CleanedModelSerializer):
    class Meta:
        model = FoodEvent
        fields = ("start", "end", "event", "title", "is_admin")

    event = serializers.PrimaryKeyRelatedField(read_only=True)
    is_admin = serializers.SerializerMethodField("_is_admin")

    def _is_admin(self, instance):
        member = self.context["request"].member
        return can_change_order(member, instance)


class OrderSerializer(CleanedModelSerializer):
    class Meta:
        model = FoodOrder
        fields = ("pk", "payment", "product", "name", "member")
        read_only_fields = ("pk", "payment", "name", "member")

    payment = PaymentTypeField(
        source="payment.type", choices=Payment.PAYMENT_TYPE, read_only=True
    )


class AdminOrderSerializer(CleanedModelSerializer):
    class Meta:
        model = FoodOrder
        fields = ("pk", "payment", "product", "name", "member", "display_name")

    payment = PaymentTypeField(
        source="payment.type", choices=Payment.PAYMENT_TYPE, required=False
    )
    display_name = serializers.SerializerMethodField("_display_name")

    def _display_name(self, instance):
        if instance.member:
            return instance.member.get_full_name()
        return instance.name

    def validate(self, attrs):
        if attrs.get("member") and attrs.get("name"):
            raise ValidationError(
                {
                    "member": _("Either specify a member or a name"),
                    "name": _("Either specify a member or a name"),
                }
            )
        if not (attrs.get("member") or attrs.get("name")) and not self.partial:
            attrs["member"] = self.context["request"].member
        return super().validate(attrs)

    def create(self, validated_data: Any) -> Any:
        if "payment" in validated_data:
            del validated_data["payment"]
        return super().create(validated_data)

    def update(self, instance: Model, validated_data: Any) -> Any:
        if "payment" in validated_data:
            del validated_data["payment"]
        return super().update(instance, validated_data)