serializers.py

from rest_framework import serializers

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

from . import filters
from . import models
from .radios import registry


class FilterSerializer(serializers.Serializer):
    type = serializers.CharField(source='code')
    label = serializers.CharField()
    help_text = serializers.CharField()
    fields = serializers.ReadOnlyField()


class RadioSerializer(serializers.ModelSerializer):
    user = UserBasicSerializer(read_only=True)

    class Meta:
        model = models.Radio
        fields = (
            'id',
            'is_public',
            'name',
            'creation_date',
            'user',
            'config',
            'description')
        read_only_fields = ('user', 'creation_date')

    def save(self, **kwargs):
        kwargs['config'] = [
            filters.registry[f['type']].clean_config(f)
            for f in self.validated_data['config']
        ]

        return super().save(**kwargs)

class RadioSessionTrackSerializerCreate(serializers.ModelSerializer):
    class Meta:
        model = models.RadioSessionTrack
        fields = ('session',)


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

    class Meta:
        model = models.RadioSessionTrack
        fields = ('id', 'session', 'position', 'track')


class RadioSessionSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.RadioSession
        fields = (
            'id',
            'radio_type',
            'related_object_id',
            'user',
            'creation_date',
            'custom_radio',
            'session_key')

    def validate(self, data):
        registry[data['radio_type']]().validate_session(data, **self.context)
        return data

    def create(self, validated_data):
        if self.context.get('user'):
            validated_data['user'] = self.context.get('user')
        else:
            validated_data['session_key'] = self.context['session_key']
        if validated_data.get('related_object_id'):
            radio = registry[validated_data['radio_type']]()
            validated_data['related_object'] = radio.get_related_object(validated_data['related_object_id'])
        return super().create(validated_data)