diff --git a/server/proj/audio/admin.py b/server/proj/audio/admin.py index 7dd82aa..59b465e 100644 --- a/server/proj/audio/admin.py +++ b/server/proj/audio/admin.py @@ -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') diff --git a/server/proj/audio/views.py b/server/proj/audio/views.py index d6649cd..5ac821d 100644 --- a/server/proj/audio/views.py +++ b/server/proj/audio/views.py @@ -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) \ No newline at end of file + 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) + \ No newline at end of file