20 lines
597 B
Python
20 lines
597 B
Python
from rest_framework import generics
|
|
|
|
from . import models, serializers
|
|
|
|
|
|
class ImageBaseView(generics.CreateAPIView):
|
|
"""Upload image to gallery."""
|
|
model = models.Image
|
|
queryset = models.Image.objects.all()
|
|
serializer_class = serializers.ImageSerializer
|
|
|
|
|
|
class NewsImageListView(ImageBaseView, generics.ListAPIView):
|
|
"""Return list of uploaded images for news object."""
|
|
|
|
def get_queryset(self):
|
|
"""Override get_queryset method."""
|
|
qs = super(NewsImageListView, self).get_queryset()
|
|
return qs.filter(news_gallery__news=self.kwargs.get('news_id'))
|