добавил возможность создания избранных треков
This commit is contained in:
parent
64f23118cd
commit
59ccea4184
|
|
@ -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')
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
Loading…
Reference in New Issue