72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
from rest_framework import generics
|
|
from rest_framework import permissions
|
|
|
|
from collection import models
|
|
from collection.serializers import common as serializers
|
|
|
|
|
|
# Mixins
|
|
class CollectionViewMixin(generics.GenericAPIView):
|
|
"""Mixin for Collection view"""
|
|
model = models.Collection
|
|
queryset = models.Collection.objects.all()
|
|
|
|
|
|
class CollectionItemViewMixin(generics.GenericAPIView):
|
|
"""Mixin for CollectionItem view"""
|
|
model = models.CollectionItem
|
|
queryset = models.CollectionItem.objects.all()
|
|
|
|
|
|
class GuideViewMixin(generics.GenericAPIView):
|
|
"""Mixin for Guide view"""
|
|
model = models.Guide
|
|
queryset = models.Guide.objects.all()
|
|
|
|
|
|
# Views
|
|
# Collections
|
|
class CollectionListView(CollectionViewMixin, generics.ListAPIView):
|
|
"""List Collection view"""
|
|
pagination_class = None
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.CollectionSerializer
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method"""
|
|
return models.Collection.objects.published()\
|
|
.by_country_code(code=self.request.country_code)\
|
|
.order_by('-on_top', '-created')
|
|
|
|
|
|
class CollectionRetrieveView(CollectionViewMixin, generics.RetrieveAPIView):
|
|
"""Retrieve Collection view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.CollectionSerializer
|
|
|
|
|
|
# CollectionItem
|
|
class CollectionItemListView(CollectionItemViewMixin, generics.ListAPIView):
|
|
"""List CollectionItem view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.CollectionItemSerializer
|
|
|
|
|
|
class CollectionItemRetrieveView(CollectionItemViewMixin, generics.RetrieveAPIView):
|
|
"""Retrieve CollectionItem view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.CollectionItemSerializer
|
|
|
|
|
|
# Guide
|
|
class GuideListView(GuideViewMixin, generics.ListAPIView):
|
|
"""List Guide view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.GuideSerializer
|
|
|
|
|
|
class GuideRetrieveView(GuideViewMixin, generics.RetrieveAPIView):
|
|
"""Retrieve Guide view"""
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = serializers.GuideSerializer
|