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'))