from ..models import Article, Program, StaticPage
from .page import PageDetailView, PageListView


__all__ = ['ArticleDetailView', 'ArticleListView']


class ArticleDetailView(PageDetailView):
    has_sidebar = True
    model = Article

    def get_sidebar_queryset(self):
        qs = Article.objects.select_related('cover') \
                    .published().filter(is_static=False) \
                    .order_by('-pub_date')
        return qs


class ArticleListView(PageListView):
    model = Article
    has_headline = True
    is_static = False
    parent_model = Program
    attach_to_value = StaticPage.ATTACH_TO_ARTICLES

    def get_queryset(self):
        return super().get_queryset().filter(is_static=self.is_static)