from rest_framework import serializers
from .models import Song, FavoriteSong

class SongSerializer(serializers.ModelSerializer):
    class Meta:
        model = Song    
        fields = ('id', 'unique_id', 'azura_id', 'title', 'artist', 'album', 'genre', 'art')

class FavoriteSongSerializer(serializers.ModelSerializer):
    class Meta:
        model = FavoriteSong    
        fields = ('id', 'song', 'user')

    def to_representation(self, instance):
        rep = super().to_representation(instance)
        rep["song"] = SongSerializer(
            instance.song, many=True).data
        return rep