from rest_framework import serializers
from .models import Wrestler


class WrestlerSerializer(serializers.ModelSerializer):
    club_name = serializers.CharField(source='club.name', read_only=True)
    age = serializers.SerializerMethodField()
    
    class Meta:
        model = Wrestler
        fields = '__all__'
        read_only_fields = ['created_at', 'updated_at', 'age']
    
    def get_age(self, obj):
        return obj.calculate_age()