articles.py

from django.http import HttpResponseRedirect
from django.views.generic.detail import DetailView
from django.views.generic.edit import UpdateView
from django.utils.translation import ugettext_lazy as _
from django.urls import reverse, reverse_lazy
from django import forms

from crispy_forms.helper import FormHelper
from crispy_forms.layout import Layout, Field, Div, HTML
from crispy_forms.bootstrap import AppendedText

from taggit.models import Tag
from rp.forms import TagMultipleChoiceField

from rp.models import Article
from .votes import UDList


class ArticleListFlux(UDList):
    model = Article
    paginate_by = 10

    def get_queryset(self):
        filter_view = self.kwargs.get("filter_view", "draft")

        if filter_view == "published":
            qs = Article.objects.filter(status="PUBLISHED")
        elif filter_view == "draft":
            qs = Article.objects.extra(where=[
                "(und_score_up + und_score_down >= 3) or priority",
                "status='PENDING'"
            ])
        elif filter_view == "rejected":
            qs = Article.objects.filter(status="REJECTED")
        else:
            qs = Article.objects.filter(status="PENDING")

        return qs.order_by('-created_at')

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["filter_view"] = self.kwargs.get("filter_view", "draft")
        context["nb_draft"] = Article.objects.extra(where=[
            "(und_score_up + und_score_down >= 3) or priority",
            "status='PENDING'"
        ]).count()
        return context


class ArticleDetailView(DetailView):
    model = Article
    preview = False

    def get_context_data(self, **kwargs):
        context = super(ArticleDetailView, self).get_context_data(**kwargs)
        context['is_preview'] = self.preview
        return context


class ArticleEdit(UpdateView):
    model = Article
    fields = ['screenshot', 'url', 'lang', 'title', 'tags', 'extracts']
    success_url = reverse_lazy("rp:article-list")

    def form_valid(self, form):
        self.object = form.save()

        if "preview" in self.request.POST:
            self.success_url = reverse("rp:article-preview", args=[self.object.id])
        elif "view" in self.request.POST:
            self.success_url = reverse("rp:article-view", args=[self.object.id])
        elif "publish" in self.request.POST:
            self.object.publish()

        return HttpResponseRedirect(self.get_success_url())

    def get_form(self, form_class=None):
        form = super().get_form(form_class)
        form.fields["screenshot"].widget = forms.widgets.FileInput()

        form.helper = FormHelper()
        form.helper.form_tag = False

        # Change the layout of the tags field
        form.fields["tags"] = TagMultipleChoiceField(
            queryset=Tag.objects.all(), required=False)

        left_layout = Div(
            Div(
                Field('title', wrapper_class='col-sm-10'),
                Field('lang', wrapper_class='col-sm-2'),
                css_class="row"),
            AppendedText(
                'url',
                _('<a href="%s">Go to link</a>') % form.initial['url']),
            'tags',
            'extracts',
            css_class="col-sm-8")

        right_layout = Div(
            'screenshot',
            HTML(
                """
                {% if form.screenshot.value %}
                <img class="img-responsive mb-4"
                     src="/media/{{ form.screenshot.value }}">
                {% endif %}
                """
            ),
            css_class="col-sm-4")

        form.helper.layout = Layout(
            Div(left_layout, right_layout, css_class="row"))

        return form