from rest_framework import serializers
from .models import Club


class ClubSerializer(serializers.ModelSerializer):
    wrestler_count = serializers.SerializerMethodField()
    
    class Meta:
        model = Club
        fields = '__all__'
        read_only_fields = ['created_at', 'updated_at']
    
    def get_wrestler_count(self, obj):
        return obj.wrestlers.filter(is_active=True).count()