66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
from rest_framework import generics
|
|
from rest_framework import permissions
|
|
from rest_framework import viewsets
|
|
from rest_framework.response import Response
|
|
|
|
from collection import models
|
|
from collection.serializers import back as serializers
|
|
from utils.views import BindObjectMixin
|
|
|
|
|
|
class CollectionViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
|
|
"""ViewSet for Collection model."""
|
|
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
queryset = models.Collection.objects.all()
|
|
serializer_class = serializers.CollectionBackOfficeSerializer
|
|
|
|
|
|
class CollectionBackOfficeViewSet(mixins.CreateModelMixin,
|
|
mixins.UpdateModelMixin,
|
|
mixins.DestroyModelMixin,
|
|
mixins.RetrieveModelMixin,
|
|
BindObjectMixin,
|
|
CollectionViewSet):
|
|
"""ViewSet for Collection model for BackOffice users."""
|
|
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
queryset = models.Collection.objects.all()
|
|
serializer_class = serializers.CollectionBackOfficeSerializer
|
|
bind_object_serializer_class = serializers.CollectionBindObjectSerializer
|
|
|
|
def perform_binding(self, serializer):
|
|
data = serializer.validated_data
|
|
collection = data.pop('collection')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT:
|
|
collection.establishments.add(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.PRODUCT:
|
|
collection.products.add(related_object)
|
|
|
|
def perform_unbinding(self, serializer):
|
|
data = serializer.validated_data
|
|
collection = data.pop('collection')
|
|
obj_type = data.get('type')
|
|
related_object = data.get('related_object')
|
|
if obj_type == self.bind_object_serializer_class.ESTABLISHMENT:
|
|
collection.establishments.remove(related_object)
|
|
elif obj_type == self.bind_object_serializer_class.PRODUCT:
|
|
collection.products.remove(related_object)
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
queryset = self.filter_queryset(self.get_queryset()).order_by('rank', '-start')
|
|
serializer = self.get_serializer(queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class CollectionBackOfficeView(generics.GenericAPIView):
|
|
"""ViewS for Collection model."""
|
|
|
|
pagination_class = None
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
queryset = models.Collection.objects.all()
|
|
serializer_class = serializers.CollectionBackOfficeSerializer
|