from rest_framework import serializers
from .models import *

class EquipmentSerializer(serializers.ModelSerializer):   
    class Meta:
        model = Equipment
        fields = '__all__'


class WorkoutEquipmentSerializer(serializers.ModelSerializer):    
    name = serializers.SerializerMethodField()

    class Meta:
        model = WorkoutEquipment
        fields = '__all__'

    def get_name(self, obj):
        return obj.equipment.name