59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from django.shortcuts import render
|
|
from rest_framework.viewsets import ViewSet
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import action
|
|
from rest_framework import status
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from .schemas import SongSchema, DeleteSongSchema
|
|
from .models import Song, FavoriteSong
|
|
from .serializers import SongSerializer, FavoriteSongSerializer
|
|
|
|
class SongViewSet(ViewSet):
|
|
def list(self, request):
|
|
queryset = Song.objects.all()
|
|
serializer = SongSerializer(queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(detail=False, methods=['post'], schema=SongSchema())
|
|
def create_song(self, request):
|
|
|
|
""" song = Song.objects.filter(
|
|
azura_id = request.data['azura_id'],
|
|
title = request.data['title'],
|
|
artist = request.data['artist'],
|
|
album = request.data['album'],
|
|
genre = request.data['genre'],
|
|
art = request.data['art']
|
|
)
|
|
|
|
if song.exists():
|
|
data = {
|
|
"song": serializer.data['id'],
|
|
"user": request.user.pk
|
|
} """
|
|
|
|
|
|
serializer = SongSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
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)
|
|
|
|
|
|
@action(detail=False, methods=['post'], schema=DeleteSongSchema())
|
|
def delete_song(self, request):
|
|
song_id = request.data['song_id']
|
|
item = get_object_or_404(FavoriteSong, pk=song_id)
|
|
item.delete()
|
|
return Response(status=status.HTTP_202_ACCEPTED)
|
|
|
|
|