добавил возможность создания избранных треков

This commit is contained in:
Mike0001-droid 2024-06-06 18:52:57 +05:00
parent 64f23118cd
commit 59ccea4184
2 changed files with 17 additions and 5 deletions

View File

@ -1,6 +1,10 @@
from django.contrib import admin
from .models import Song
from .models import Song, FavoriteSong
@admin.register(Song)
class SongAdmin(admin.ModelAdmin):
list_display = ('title', 'artist', 'genre', 'album')
@admin.register(Song)
class FavoriteSongAdmin(admin.ModelAdmin):
list_display = ('id', 'song', 'user')

View File

@ -7,7 +7,7 @@ from django.shortcuts import get_object_or_404
from .schemas import SongSchema
from .models import Song
from .serializers import SongSerializer
from .serializers import SongSerializer, FavoriteSongSerializer
class SongViewSet(ViewSet):
def list(self, request):
@ -20,6 +20,14 @@ class SongViewSet(ViewSet):
serializer = SongSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
track_id = serializer.data['id']
track = get_object_or_404(Song, pk=track_id)
return Response(f"{track}", status=status.HTTP_201_CREATED)
data = {
"song": serializer.data['id'],
"user": request.user.pk
}
serializer_1 = FavoriteSongSerializer(data=data)
if serializer_1.is_valid():
serializer_1.save()
return Response(serializer_1.data, status=status.HTTP_201_CREATED)
else:
return Response(serializer_1.errors, status=status.HTTP_400_BAD_REQUEST)