spradio-server-django/savepointradio/api/views/profiles.py

79 lines
2.9 KiB
Python
Raw Normal View History

2018-04-05 14:59:30 +00:00
from django.shortcuts import get_object_or_404
from rest_framework import mixins, status, viewsets
2018-04-05 14:59:30 +00:00
from rest_framework.decorators import action
from rest_framework.permissions import AllowAny
2018-04-05 14:59:30 +00:00
from rest_framework.response import Response
from profiles.models import RadioProfile, SongRequest
2018-04-05 14:59:30 +00:00
from ..permissions import IsAdminOwnerOrReadOnly
from ..serializers.profiles import (BasicProfileSerializer,
FullProfileSerializer,
2018-04-09 16:11:15 +00:00
HistorySerializer,
BasicProfileRatingsSerializer)
from ..serializers.radio import SongListSerializer
2018-04-05 14:59:30 +00:00
class ProfileViewSet(viewsets.ModelViewSet):
permission_classes = [IsAdminOwnerOrReadOnly]
queryset = RadioProfile.objects.all()
serializer_class = BasicProfileSerializer
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.is_owner = False
def get_serializer_class(self):
'''
Choose a different serializer based on if the requesting user is an
admin or is the same person as the profile requested.
'''
if self.request.user.is_staff or self.is_owner:
return FullProfileSerializer
return BasicProfileSerializer
def get_object(self):
'''
Grab the object as normal, but let us know if the requesting user is
the owner.
'''
obj = get_object_or_404(self.get_queryset(), **self.kwargs)
if self.request.user.pk == obj.user.pk:
self.is_owner = True
else:
self.is_owner = False
self.check_object_permissions(self.request, obj)
return obj
2018-04-06 16:24:13 +00:00
@action(detail=True, permission_classes=[AllowAny])
def favorites(self, request, pk=None):
profile = self.get_object()
favorites = profile.favorites.all().order_by('sorted_title')
page = self.paginate_queryset(favorites)
if page is not None:
serializer = SongListSerializer(page, many=True)
2018-04-06 16:24:13 +00:00
return self.get_paginated_response(serializer.data)
serializer = SongListSerializer(favorites, many=True)
2018-04-06 16:24:13 +00:00
return Response(serializer.data)
@action(detail=True, permission_classes=[AllowAny])
def ratings(self, request, pk=None):
2018-04-06 16:24:13 +00:00
profile = self.get_object()
ratings = profile.rating_profile.all().order_by('-created_date')
2018-04-06 16:24:13 +00:00
page = self.paginate_queryset(ratings)
if page is not None:
serializer = BasicProfileRatingsSerializer(page, many=True)
return self.get_paginated_response(serializer.data)
2018-04-06 16:24:13 +00:00
serializer = BasicProfileRatingsSerializer(ratings, many=True)
return Response(serializer.data)
2018-04-06 16:24:13 +00:00
class HistoryViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
permission_classes = [AllowAny]
queryset = SongRequest.objects.all()
serializer_class = HistorySerializer