diff --git a/src/lyrics/serializers.py b/src/lyrics/serializers.py index 47d4398..9e0aaa2 100644 --- a/src/lyrics/serializers.py +++ b/src/lyrics/serializers.py @@ -1,3 +1,5 @@ +import os + from rest_framework import serializers from .models import Artista, Album, Song @@ -7,6 +9,12 @@ class ArtistaSerializer(serializers.ModelSerializer): model = Artista fields = '__all__' # Incluir todos los campos del modelo + def to_representation(self, instance): + ret = super().to_representation(instance) + ret['foto'] = "artistas/" + os.path.basename(ret['foto']) + + return ret + class AlbumSerializer(serializers.ModelSerializer): artista_nombre = serializers.CharField(source='artist.nombre', read_only=True) @@ -15,6 +23,12 @@ class AlbumSerializer(serializers.ModelSerializer): model = Album fields = ['name', 'year', 'cover_image', 'artista_nombre'] + def to_representation(self, instance): + ret = super().to_representation(instance) + ret['cover_image'] = "cover_image/" + os.path.basename(ret['cover_image']) + + return ret + class CancionSerializer(serializers.ModelSerializer): artista_nombre = serializers.CharField(source='artist.nombre', read_only=True)