ITRadio/server/proj/audio/views.py

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, AllowAny
import requests
from django.http import HttpResponse

from .schemas import SongSchema, DeleteSongSchema, PlayListSchema
from .models import Song, FavoriteSong, PlayList
from .serializers import SongSerializer, FavoriteSongSerializer, PlayListSerializer

class PlayListViewSet(GenericViewSet):
    queryset = PlayList
    serializer_class = PlayListSerializer
    permission_classes = (IsAuthenticated,)
    def list(self, request):
        queryset = self.get_queryset().objects.filter(user=request.user)
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data, status=status.HTTP_200_OK)
    
    def retrieve(self, request, pk):
        try:
            queryset = self.get_queryset().objects.get(pk=pk, user=request.user)
            serializer = self.get_serializer(queryset)
            return Response(serializer.data, status=status.HTTP_200_OK)
        except ObjectDoesNotExist:
            return Response(
                {'detail': 'Объекта не существует', 'error': {'PlayList': 'Объекта не существует'}}, 
            status=status.HTTP_404_NOT_FOUND)
        
    @action(detail=False, methods=['post'], schema=PlayListSchema())
    def create_playlist(self, request):
        
        if request.data.get('name', False):
            try:
                instance = PlayList.objects.get(pk=request.data['playlist_id'], user=request.user.pk)
                data = request.data
                data.update(user=request.user.pk)
                serializer = self.get_serializer(data=request.data, instance=instance)
                serializer.is_valid(raise_exception=True)
                serializer.save() 
                return Response(serializer.data)
            except ObjectDoesNotExist:
                return Response(
                    {'detail': 'Объекта не существует', 'error': {'PlayList': 'Объекта не существует'}}, 
                status=status.HTTP_404_NOT_FOUND)
        else:
            playlist_pk = self.get_queryset().objects.filter(user=request.user.pk)
            number_playlist = len(playlist_pk) + 1
            name = f"Плейлист № {number_playlist}"
            data = request.data
            data.update(user=request.user.pk, name=name)
            serializer = self.get_serializer(data=request.data)
            serializer.is_valid(raise_exception=True)
            serializer.save() 
            return Response(serializer.data)
        
    @action(detail=False, methods=['post'], schema=PlayListSchema())
    def delete_playlist(self, request):
        try:
            item = PlayList.objects.get(user=request.user, pk=request.data.get('playlist_id'))
            item.delete()
            return Response(status=status.HTTP_202_ACCEPTED)
        except ObjectDoesNotExist:
            return Response(
                {'detail': 'Объекта не существует', 'error': {'PlayList': 'Объекта не существует'}}, 
            status=status.HTTP_404_NOT_FOUND)
        
    @action(detail=False, methods=['post'], schema=PlayListSchema())
    def add_to_playlist(self, request):
        song = None
        try:
            song = Song.objects.get(azura_id=request.data.get('azura_id')).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)
            song_serializer = SongSerializer(data=data)
            if song_serializer.is_valid():
                song_serializer.save()
            song = song_serializer.data.get('id')
        try:
            instance = PlayList.objects.get(pk=request.data.get('playlist_id'))
        except ObjectDoesNotExist:
            return Response(
                {'detail': 'Плейлиста не существует', 'error': {'PlayList': 'Плейлиста не существует'}}, 
            status=status.HTTP_404_NOT_FOUND)
        
        songs = list(instance.song.all().values_list('pk', flat=True)) + [song]
        data = {
            'playlist_id': request.data.get('playlist_id'),
            'song': songs
        }
        serializer = PlayListSerializer(data=data, partial=True, instance=instance)
        serializer.is_valid(raise_exception=True)
        serializer.save() 
        return Response(serializer.data)
        
    
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)
            return Response({"is_favorite": True}, status=status.HTTP_200_OK)
        except ObjectDoesNotExist:
            return Response({"is_favorite": False}, status=status.HTTP_200_OK)
        
    @action(
            detail=False,
            methods=['get'],
            url_path='get_audio/(?P<azura_id>[a-zA-Z0-9_]+)',
            url_name='get_audio',
        )
    def get_audio(self, request, azura_id):
        try:
            song_obj = Song.objects.get(azura_id=azura_id)
            file_url = song_obj.unique_id
            API_KEY = "49226d3488aac3f5:18d88659c6c1c5e131a0ce0a94d55235"
            headers = {
                "Authorization": f"Bearer {API_KEY}"
            }
            response = requests.get(file_url, headers=headers)
            response.raise_for_status()

            file_response = HttpResponse(response.content, content_type='audio/mpeg')
            file = f'{song_obj.title}-{song_obj.artist}.mp3'
            file_response['Content-Disposition'] = f'attachment; filename={file}'
            return file_response
            
        except ObjectDoesNotExist:
            return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
        
    

    @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:
            song = Song.objects.get(azura_id=request.data.get('azura_id'))
            item = FavoriteSong.objects.get(user=request.user, song=song.pk)
            item.delete()
            return Response(status=status.HTTP_202_ACCEPTED)
        except ObjectDoesNotExist:
            return Response({"error": 'Объекта не существует'}, status=status.HTTP_404_NOT_FOUND)
        
    @action(detail=False, methods=['get'])
    def get_all_song(self, request):
        file_url = "http://82.97.242.49:10084/api/station/1/files"
        API_KEY = "49226d3488aac3f5:18d88659c6c1c5e131a0ce0a94d55235"
        headers = {
            "Authorization": f"Bearer {API_KEY}"
        }
        response = requests.get(file_url, headers=headers)
        data = []
        for i in response.json():
            i['azura_id'] = i.pop('song_id')
            data.append(i)
        return Response(data, status=status.HTTP_200_OK)