from rest_framework import serializers

from .models import Cuisine, Dish


class CuisineSerializer(serializers.ModelSerializer):
    dish_count = serializers.SerializerMethodField()

    class Meta:
        model = Cuisine
        fields = ['id', 'name', 'description', 'sort_order', 'is_active', 'dish_count', 'created_at']

    def get_dish_count(self, obj):
        return obj.dishes.count()


class DishSerializer(serializers.ModelSerializer):
    dish_type_display = serializers.CharField(source='get_dish_type_display', read_only=True)
    cuisine_name = serializers.CharField(read_only=True)

    class Meta:
        model = Dish
        fields = ['id', 'name', 'dish_type', 'dish_type_display', 'cuisine', 'cuisine_name', 'ingredient_detail', 'protein', 'fat', 'calorie', 'description', 'is_active', 'created_at']