serializers.py

from django.conf import settings
from django.db import transaction
from rest_framework import serializers
from taggit.models import Tag

from funkwhale_api.music.models import Track
from funkwhale_api.music.serializers import TrackSerializerNested
from funkwhale_api.users.serializers import UserBasicSerializer
from . import models


class PlaylistTrackSerializer(serializers.ModelSerializer):
    track = TrackSerializerNested()

    class Meta:
        model = models.PlaylistTrack
        fields = ('id', 'track', 'playlist', 'index', 'creation_date')


class PlaylistTrackWriteSerializer(serializers.ModelSerializer):
    index = serializers.IntegerField(
        required=False, min_value=0, allow_null=True)

    class Meta:
        model = models.PlaylistTrack
        fields = ('id', 'track', 'playlist', 'index')

    def validate_playlist(self, value):
        if self.context.get('request'):
            # validate proper ownership on the playlist
            if self.context['request'].user != value.user:
                raise serializers.ValidationError(
                    'You do not have the permission to edit this playlist')
        existing = value.playlist_tracks.count()
        if existing >= settings.PLAYLISTS_MAX_TRACKS:
            raise serializers.ValidationError(
                'Playlist has reached the maximum of {} tracks'.format(
                    settings.PLAYLISTS_MAX_TRACKS))
        return value

    @transaction.atomic
    def create(self, validated_data):
        index = validated_data.pop('index', None)
        instance = super().create(validated_data)
        instance.playlist.insert(instance, index)
        return instance

    @transaction.atomic
    def update(self, instance, validated_data):
        update_index = 'index' in validated_data
        index = validated_data.pop('index', None)
        super().update(instance, validated_data)
        if update_index:
            instance.playlist.insert(instance, index)
        return instance

    def get_unique_together_validators(self):
        """
        We explicitely disable unique together validation here
        because it collides with our internal logic
        """
        return []


class PlaylistSerializer(serializers.ModelSerializer):
    tracks_count = serializers.SerializerMethodField(read_only=True)
    user = UserBasicSerializer(read_only=True)

    class Meta:
        model = models.Playlist
        fields = (
            'id',
            'name',
            'tracks_count',
            'user',
            'privacy_level',)
        read_only_fields = [
            'id',
            'modification_date',
            'creation_date',]

    def get_tracks_count(self, obj):
        try:
            return obj.tracks_count
        except AttributeError:
            # no annotation?
            return obj.playlist_tracks.count()


class PlaylistAddManySerializer(serializers.Serializer):
    tracks = serializers.PrimaryKeyRelatedField(
        many=True, queryset=Track.objects.for_nested_serialization())