cancion_adv.py

from rest_framework import viewsets, status
from rest_framework.response import Response
from .models import *
from .serializers import *

class CancionesForAlbumViewSet(viewsets.ModelViewSet):
    """!
    Vistas para registrar las canciones de un album

    @author Rodrigo Boet (rudmanmrrod at gmail.com)
    @date 28-05-2019
    @version 1.0.0
    """
    model = Canciones
    queryset = Canciones.objects.all()
    serializer_class = CancionesSolasSerializer

    def get_queryset(self):
        queryset = Canciones.objects.filter(album_id=self.kwargs['id']).all()
        return queryset

    def create(self, request, id):
        request.data['album_id'] = id
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.save() 
        return Response(serializer.data, status=status.HTTP_201_CREATED)