Skip to content
Snippets Groups Projects
views.py 3.83 KiB
Newer Older
  • Learn to ignore specific revisions
  • from django.db.models import Count
    
    from django.db import transaction
    
    from rest_framework import exceptions
    
    from rest_framework import generics, mixins, viewsets
    from rest_framework import status
    
    from rest_framework.decorators import detail_route
    
    from rest_framework.response import Response
    
    from rest_framework.permissions import IsAuthenticatedOrReadOnly
    
    
    from funkwhale_api.music.models import Track
    
    from funkwhale_api.common import permissions
    from funkwhale_api.common import fields
    
    from . import models
    from . import serializers
    
    
    class PlaylistViewSet(
            mixins.RetrieveModelMixin,
            mixins.CreateModelMixin,
    
            mixins.UpdateModelMixin,
    
            mixins.DestroyModelMixin,
            mixins.ListModelMixin,
            viewsets.GenericViewSet):
    
        serializer_class = serializers.PlaylistSerializer
    
            models.Playlist.objects.all().select_related('user')
    
                  .annotate(tracks_count=Count('playlist_tracks'))
        )
    
        permission_classes = [
            permissions.ConditionalAuthentication,
            permissions.OwnerPermission,
            IsAuthenticatedOrReadOnly,
        ]
    
        owner_checks = ['write']
    
        filter_class = filters.PlaylistFilter
    
        def get_serializer_class(self):
            if self.request.method in ['PUT', 'PATCH', 'DELETE', 'POST']:
                return serializers.PlaylistWriteSerializer
            return self.serializer_class
    
    
        @detail_route(methods=['get'])
        def tracks(self, request, *args, **kwargs):
            playlist = self.get_object()
            plts = playlist.playlist_tracks.all()
            serializer = serializers.PlaylistTrackSerializer(plts, many=True)
            data = {
                'count': len(plts),
    
                'results': serializer.data
    
            }
            return Response(data, status=200)
    
    
        @detail_route(methods=['post'])
        @transaction.atomic
        def add(self, request, *args, **kwargs):
            playlist = self.get_object()
            serializer = serializers.PlaylistAddManySerializer(data=request.data)
            serializer.is_valid(raise_exception=True)
            try:
                plts = playlist.insert_many(serializer.validated_data['tracks'])
            except exceptions.ValidationError as e:
                payload = {'playlist': e.detail}
                return Response(payload, status=400)
            serializer = serializers.PlaylistTrackSerializer(plts, many=True)
            data = {
                'count': len(plts),
                'results': serializer.data
            }
            return Response(data, status=201)
    
    
            return self.queryset.filter(
                fields.privacy_level_query(self.request.user))
    
            return serializer.save(
                user=self.request.user,
                privacy_level=serializer.validated_data.get(
                    'privacy_level', self.request.user.privacy_level)
            )
    
            mixins.RetrieveModelMixin,
    
            mixins.UpdateModelMixin,
    
            mixins.DestroyModelMixin,
            mixins.ListModelMixin,
            viewsets.GenericViewSet):
    
        serializer_class = serializers.PlaylistTrackSerializer
        queryset = (models.PlaylistTrack.objects.all())
    
        permission_classes = [
            permissions.ConditionalAuthentication,
            permissions.OwnerPermission,
            IsAuthenticatedOrReadOnly,
        ]
    
        owner_field = 'playlist.user'
        owner_checks = ['write']
    
        def get_serializer_class(self):
            if self.request.method in ['PUT', 'PATCH', 'DELETE', 'POST']:
                return serializers.PlaylistTrackWriteSerializer
            return self.serializer_class
    
            return self.queryset.filter(
                fields.privacy_level_query(
                    self.request.user,
                    lookup_field='playlist__privacy_level'))
    
    
        def perform_destroy(self, instance):
            instance.delete(update_indexes=True)