Views and logic for custom radios

api/funkwhale_api/radios/views.py

@@ -1,14 +1,72 @@
+from django.db.models import Q
+from django.http import Http404
+
 from rest_framework import generics, mixins, viewsets
 from rest_framework import status
 from rest_framework.response import Response
-from rest_framework.decorators import detail_route
+from rest_framework.decorators import detail_route, list_route
 
 from funkwhale_api.music.serializers import TrackSerializerNested
 from funkwhale_api.common.permissions import ConditionalAuthentication
 
 from . import models
+from . import filters
+from . import filtersets
 from . import serializers
 
+
+class RadioViewSet(
+        mixins.CreateModelMixin,
+        mixins.RetrieveModelMixin,
+        mixins.UpdateModelMixin,
+        mixins.ListModelMixin,
+        viewsets.GenericViewSet):
+
+    serializer_class = serializers.RadioSerializer
+    permission_classes = [ConditionalAuthentication]
+    filter_class = filtersets.RadioFilter
+
+    def get_queryset(self):
+        query = Q(is_public=True)
+        if self.request.user.is_authenticated:
+            query |= Q(user=self.request.user)
+        return models.Radio.objects.filter(query)
+
+    def perform_create(self, serializer):
+        return serializer.save(user=self.request.user)
+
+    def perform_update(self, serializer):
+        if serializer.instance.user != self.request.user:
+            raise Http404
+        return serializer.save(user=self.request.user)
+
+    @list_route(methods=['get'])
+    def filters(self, request, *args, **kwargs):
+        serializer = serializers.FilterSerializer(
+            filters.registry.exposed_filters, many=True)
+        return Response(serializer.data)
+
+    @list_route(methods=['post'])
+    def validate(self, request, *args, **kwargs):
+        try:
+            f_list = request.data['filters']
+        except KeyError:
+            return Response(
+                {'error': 'You must provide a filters list'}, status=400)
+        data = {
+            'filters': []
+        }
+        for f in f_list:
+            results = filters.test(f)
+            if results['candidates']['sample']:
+                qs = results['candidates']['sample'].for_nested_serialization()
+                results['candidates']['sample'] = TrackSerializerNested(
+                    qs, many=True).data
+            data['filters'].append(results)
+
+        return Response(data)
+
+
 class RadioSessionViewSet(mixins.CreateModelMixin,
                           mixins.RetrieveModelMixin,
                           viewsets.GenericViewSet):