From 59ccea418437faad5c73a37f0701afc65fad6443 Mon Sep 17 00:00:00 2001
From: Mike0001-droid <Majorov-2004@list.ru>
Date: Thu, 6 Jun 2024 18:52:57 +0500
Subject: [PATCH] =?UTF-8?q?=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8=D0=BB=20?=
 =?UTF-8?q?=D0=B2=D0=BE=D0=B7=D0=BC=D0=BE=D0=B6=D0=BD=D0=BE=D1=81=D1=82?=
 =?UTF-8?q?=D1=8C=20=D1=81=D0=BE=D0=B7=D0=B4=D0=B0=D0=BD=D0=B8=D1=8F=20?=
 =?UTF-8?q?=D0=B8=D0=B7=D0=B1=D1=80=D0=B0=D0=BD=D0=BD=D1=8B=D1=85=20=D1=82?=
 =?UTF-8?q?=D1=80=D0=B5=D0=BA=D0=BE=D0=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/proj/audio/admin.py |  6 +++++-
 server/proj/audio/views.py | 16 ++++++++++++----
 2 files changed, 17 insertions(+), 5 deletions(-)

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