91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
from rest_framework import decorators
|
|
from rest_framework import exceptions
|
|
from rest_framework import mixins
|
|
from rest_framework import permissions as rest_permissions
|
|
from rest_framework import response
|
|
from rest_framework import viewsets
|
|
|
|
from django import http
|
|
from django.db.utils import IntegrityError
|
|
|
|
from funkwhale_api.common import permissions
|
|
from funkwhale_api.common import preferences
|
|
from funkwhale_api.users.oauth import permissions as oauth_permissions
|
|
|
|
from . import filters, models, serializers
|
|
|
|
|
|
class ChannelsMixin(object):
|
|
def dispatch(self, request, *args, **kwargs):
|
|
if not preferences.get("audio__channels_enabled"):
|
|
return http.HttpResponse(status=405)
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
|
|
class ChannelViewSet(
|
|
ChannelsMixin,
|
|
mixins.CreateModelMixin,
|
|
mixins.RetrieveModelMixin,
|
|
mixins.UpdateModelMixin,
|
|
mixins.ListModelMixin,
|
|
mixins.DestroyModelMixin,
|
|
viewsets.GenericViewSet,
|
|
):
|
|
lookup_field = "uuid"
|
|
filterset_class = filters.ChannelFilter
|
|
serializer_class = serializers.ChannelSerializer
|
|
queryset = (
|
|
models.Channel.objects.all()
|
|
.prefetch_related("library", "attributed_to", "artist__description", "actor")
|
|
.order_by("-creation_date")
|
|
)
|
|
permission_classes = [
|
|
oauth_permissions.ScopePermission,
|
|
permissions.OwnerPermission,
|
|
]
|
|
required_scope = "libraries"
|
|
anonymous_policy = "setting"
|
|
owner_checks = ["write"]
|
|
owner_field = "attributed_to.user"
|
|
owner_exception = exceptions.PermissionDenied
|
|
|
|
def get_serializer_class(self):
|
|
if self.request.method.lower() in ["head", "get", "options"]:
|
|
return serializers.ChannelSerializer
|
|
elif self.action in ["update", "partial_update"]:
|
|
return serializers.ChannelUpdateSerializer
|
|
return serializers.ChannelCreateSerializer
|
|
|
|
def perform_create(self, serializer):
|
|
return serializer.save(attributed_to=self.request.user.actor)
|
|
|
|
@decorators.action(
|
|
detail=True,
|
|
methods=["post"],
|
|
permission_classes=[rest_permissions.IsAuthenticated],
|
|
)
|
|
def subscribe(self, request, *args, **kwargs):
|
|
object = self.get_object()
|
|
try:
|
|
subscription = object.actor.received_follows.create(
|
|
approved=True, actor=request.user.actor,
|
|
)
|
|
except IntegrityError:
|
|
# there's already a subscription for this actor/channel
|
|
subscription = object.actor.received_follows.filter(
|
|
actor=request.user.actor
|
|
).get()
|
|
|
|
data = serializers.SubscriptionSerializer(subscription).data
|
|
return response.Response(data, status=201)
|
|
|
|
@decorators.action(
|
|
detail=True,
|
|
methods=["post", "delete"],
|
|
permission_classes=[rest_permissions.IsAuthenticated],
|
|
)
|
|
def unsubscribe(self, request, *args, **kwargs):
|
|
object = self.get_object()
|
|
request.user.actor.emitted_follows.filter(target=object.actor).delete()
|
|
return response.Response(status=204)
|