Improve weekmail

com/views.py

@@ -1,18 +1,19 @@
-from django.shortcuts import render, redirect
+from django.shortcuts import render, redirect, get_object_or_404
 from django.views.generic import ListView, DetailView, RedirectView
-from django.views.generic.edit import UpdateView, CreateView
+from django.views.generic.edit import UpdateView, CreateView, DeleteView
 from django.views.generic.detail import SingleObjectMixin
 from django.utils.translation import ugettext_lazy as _
 from django.core.urlresolvers import reverse, reverse_lazy
 from django.core.exceptions import ValidationError
 from django.utils import timezone
 from django.conf import settings
+from django.db.models import Max
 from django import forms
 
 from datetime import timedelta
 
 from com.models import Sith, News, NewsDate, Weekmail, WeekmailArticle
-from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, TabedViewMixin, CanCreateMixin
+from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, TabedViewMixin, CanCreateMixin, QuickNotifMixin
 from core.views.forms import SelectDateTime
 from core.models import Notification, RealGroup
 from club.models import Club
@@ -220,7 +221,7 @@ class NewsDetailView(CanViewMixin, DetailView):
 
 # Weekmail
 
-class WeekmailEditView(UpdateView):
+class WeekmailEditView(QuickNotifMixin, UpdateView):
     model = Weekmail
     template_name = 'com/weekmail.jinja'
     fields = ['title', 'intro', 'joke', 'protip', 'conclusion']
@@ -231,6 +232,79 @@ class WeekmailEditView(UpdateView):
         if not weekmail.title:
             now = timezone.now()
             weekmail.title = _("Weekmail of the ") + (now + timedelta(days=6 - now.weekday())).strftime('%d/%m/%Y')
+        print(self.quick_notif_list)
         return weekmail
 
+    def get(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        print(self.quick_notif_list)
+        if 'up_article' in request.GET.keys():
+            art = get_object_or_404(WeekmailArticle, id=request.GET['up_article'], weekmail=self.object)
+            prev_art = self.object.articles.order_by('rank').filter(rank__lt=art.rank).last()
+            if prev_art:
+                art.rank, prev_art.rank = prev_art.rank, art.rank
+                art.save()
+                prev_art.save()
+                self.quick_notif_list += ['qn_success']
+        if 'down_article' in request.GET.keys():
+            art = get_object_or_404(WeekmailArticle, id=request.GET['down_article'], weekmail=self.object)
+            next_art = self.object.articles.order_by('rank').filter(rank__gt=art.rank).first()
+            if next_art:
+                art.rank, next_art.rank = next_art.rank, art.rank
+                art.save()
+                next_art.save()
+                self.quick_notif_list += ['qn_success']
+        if 'add_article' in request.GET.keys():
+            art = get_object_or_404(WeekmailArticle, id=request.GET['add_article'], weekmail=None)
+            art.weekmail = self.object
+            art.rank = self.object.articles.aggregate(Max('rank'))['rank__max'] or 0
+            art.rank += 1
+            art.save()
+            self.quick_notif_list += ['qn_success']
+        if 'del_article' in request.GET.keys():
+            art = get_object_or_404(WeekmailArticle, id=request.GET['del_article'], weekmail=self.object)
+            art.weekmail = None
+            art.rank = -1
+            art.save()
+            self.quick_notif_list += ['qn_success']
+        print(self.quick_notif_list)
+        return super(WeekmailEditView, self).get(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        """Add orphan articles """
+        kwargs = super(WeekmailEditView, self).get_context_data(**kwargs)
+        kwargs['orphans'] = WeekmailArticle.objects.filter(weekmail=None)
+        return kwargs
+
+class WeekmailArticleEditView(QuickNotifMixin, UpdateView):
+    """Edit an article"""
+    model = WeekmailArticle
+    fields = ['title', 'content']
+    pk_url_kwarg = "article_id"
+    template_name = 'core/edit.jinja'
+    success_url = reverse_lazy('com:weekmail')
+    quick_notif_url_arg = "qn_weekmail_article_edit"
+
+class WeekmailArticleCreateView(QuickNotifMixin, CreateView):
+    """Post an article"""
+    model = WeekmailArticle
+    fields = ['title', 'content']
+    template_name = 'core/create.jinja'
+    success_url = reverse_lazy('core:user_tools')
+    quick_notif_url_arg = "qn_weekmail_new_article"
+
+    def form_valid(self, form):
+        club = get_object_or_404(Club, id=self.kwargs['club_id'])
+        form.instance.club = club
+        form.instance.author = self.request.user
+        return super(WeekmailArticleCreateView, self).form_valid(form)
+
+class WeekmailArticleDeleteView(DeleteView):
+    """Delete an article"""
+    model = WeekmailArticle
+    template_name = 'core/delete_confirm.jinja'
+
+
+
+