86 lines
3.5 KiB
Python
86 lines
3.5 KiB
Python
from django.shortcuts import render
|
|
from rest_framework.viewsets import ViewSet, GenericViewSet
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import action
|
|
from rest_framework import status
|
|
from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned
|
|
from django.shortcuts import get_object_or_404, get_list_or_404
|
|
from rest_framework.permissions import IsAuthenticated
|
|
import requests
|
|
|
|
from .schemas import SongSchema, DeleteSongSchema
|
|
from .models import Song, FavoriteSong
|
|
from .serializers import SongSerializer, FavoriteSongSerializer
|
|
|
|
class SongViewSet(GenericViewSet):
|
|
queryset = Song.objects.all()
|
|
serializer_class = SongSerializer
|
|
permission_classes = (IsAuthenticated,)
|
|
|
|
def list(self, request):
|
|
songs_pk = FavoriteSong.objects.filter(user=request.user, song__isnull=False).values_list('song_id', flat=True)
|
|
songs = Song.objects.filter(id__in=songs_pk)
|
|
serializer = SongSerializer(songs, many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(
|
|
detail=False,
|
|
methods=['get'],
|
|
url_path='check_is_favorite/(?P<azura_id>[a-zA-Z0-9_]+)',
|
|
url_name='check_is_favorite',
|
|
)
|
|
def check_is_favorite(self, request, azura_id):
|
|
try:
|
|
song_obj = Song.objects.get(azura_id=azura_id)
|
|
favorite_songs = FavoriteSong.objects.get(user=request.user, song=song_obj)
|
|
except ObjectDoesNotExist:
|
|
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
|
|
serializer = FavoriteSongSerializer(favorite_songs)
|
|
return Response(serializer.data)
|
|
|
|
|
|
@action(detail=False, methods=['post'], schema=SongSchema())
|
|
def add_favorite(self, request):
|
|
data = None
|
|
try:
|
|
song = Song.objects.get(azura_id=request.data.get('azura_id'))
|
|
data = {
|
|
"song": song.pk,
|
|
"user": request.user.pk
|
|
}
|
|
except ObjectDoesNotExist:
|
|
file_url = f"http://82.97.242.49:10084/api/station/it-radio/file/{request.data['azura_id']}"
|
|
API_KEY = "49226d3488aac3f5:18d88659c6c1c5e131a0ce0a94d55235"
|
|
headers = {
|
|
"Authorization": f"Bearer {API_KEY}"
|
|
}
|
|
response = requests.get(file_url, headers=headers)
|
|
data = request.data
|
|
file_play = f"http://82.97.242.49:10084/api/station/it-radio/file/{response.json()['unique_id']}/play"
|
|
data.update(unique_id=file_play)
|
|
serializer = SongSerializer(data=data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
data = {
|
|
"song": serializer.data['id'],
|
|
"user": request.user.pk
|
|
}
|
|
serializer = FavoriteSongSerializer(data=data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
else:
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
@action(detail=False, methods=['post'], schema=DeleteSongSchema())
|
|
def delete_song(self, request):
|
|
try:
|
|
item = FavoriteSong.objects.get(user=request.user, song=request.data.get('song_id'))
|
|
item.delete()
|
|
return Response(status=status.HTTP_202_ACCEPTED)
|
|
except ObjectDoesNotExist:
|
|
return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
|
|
|
|
|
|
|
|
|