116 lines
4.5 KiB
Python
116 lines
4.5 KiB
Python
from rest_framework import status, viewsets
|
|
from rest_framework.decorators import action
|
|
from rest_framework.permissions import AllowAny, IsAdminUser
|
|
from rest_framework.response import Response
|
|
|
|
from profiles.models import RadioProfile, Rating
|
|
from radio.models import Album, Artist, Game, Song
|
|
from ..permissions import IsAdminOrReadOnly, IsAuthenticatedAndNotDJ
|
|
from ..serializers.profiles import (BasicSongRatingsSerializer,
|
|
RateSongSerializer)
|
|
from ..serializers.radio import (AlbumSerializer, ArtistSerializer,
|
|
GameSerializer, FullSongSerializer,
|
|
SongArtistsListSerializer,
|
|
FullSongRetrieveSerializer)
|
|
|
|
|
|
class AlbumViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAdminOrReadOnly]
|
|
queryset = Album.objects.all()
|
|
serializer_class = AlbumSerializer
|
|
|
|
|
|
class ArtistViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAdminOrReadOnly]
|
|
queryset = Artist.objects.all()
|
|
serializer_class = ArtistSerializer
|
|
|
|
|
|
class GameViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAdminOrReadOnly]
|
|
queryset = Game.objects.all()
|
|
serializer_class = GameSerializer
|
|
|
|
|
|
class SongViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAdminOrReadOnly]
|
|
queryset = Song.objects.all()
|
|
|
|
def get_serializer_class(self):
|
|
'''
|
|
Choose a different serializer based on what action is chosen.
|
|
|
|
(Thanks to https://stackoverflow.com/questions/22616973/)
|
|
'''
|
|
if self.action in ['list', 'retrieve']:
|
|
return FullSongRetrieveSerializer
|
|
return FullSongSerializer
|
|
|
|
def _artists_change(self, request, remove=False):
|
|
song = self.get_object()
|
|
serializer = SongArtistsListSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
artists = Artist.objects.filter(pk__in=serializer.data['artists'])
|
|
for artist in artists:
|
|
if remove:
|
|
song.artists.remove(artist)
|
|
else:
|
|
song.artists.add(artist)
|
|
song.save()
|
|
message = 'Artists {} song.'.format(('added to',
|
|
'removed from')[remove])
|
|
return Response({'detail': message})
|
|
else:
|
|
return Response(serializer.errors,
|
|
status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
@action(methods=['post'], detail=True, permission_classes=[IsAdminUser])
|
|
def artists_add(self, request, pk=None):
|
|
return self._artists_change(request)
|
|
|
|
@action(methods=['post'], detail=True, permission_classes=[IsAdminUser])
|
|
def artists_remove(self, request, pk=None):
|
|
return self._artists_change(request, remove=True)
|
|
|
|
@action(detail=True, permission_classes=[AllowAny])
|
|
def ratings(self, request, pk=None):
|
|
song = self.get_object()
|
|
ratings = song.rating_set.all().order_by('-created_date')
|
|
|
|
page = self.paginate_queryset(ratings)
|
|
if page is not None:
|
|
serializer = BasicSongRatingsSerializer(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
serializer = BasicSongRatingsSerializer(ratings, many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(methods=['post'],
|
|
detail=True,
|
|
permission_classes=[IsAuthenticatedAndNotDJ])
|
|
def rate(self, request, pk=None):
|
|
song = self.get_object()
|
|
serializer = RateSongSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
profile = RadioProfile.objects.get(user=request.user)
|
|
|
|
if serializer.data['rating'] == 0:
|
|
rating = song.rating_set.filter(profile=profile)
|
|
if rating:
|
|
rating.delete()
|
|
return Response({'detail': 'Rating deleted from song.'})
|
|
message = 'Cannot delete nonexistant rating.'
|
|
return Response({'detail': message},
|
|
status=status.HTTP_400_BAD_REQUEST)
|
|
else:
|
|
rating, created = Rating.objects.update_or_create(
|
|
profile=profile,
|
|
song=song,
|
|
defaults={'value': serializer.data['rating']}
|
|
)
|
|
message = 'Rating {} song.'.format(('updated for',
|
|
'created for')[created])
|
|
return Response({'detail': message})
|
|
else:
|
|
return Response(serializer.errors,
|
|
status=status.HTTP_400_BAD_REQUEST)
|