# -*- coding:utf-8 -* # # Copyright 2016,2017 # - Skia # # Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM, # http://ae.utbm.fr. # # This program is free software; you can redistribute it and/or modify it under # the terms of the GNU General Public License a published by the Free Software # Foundation; either version 3 of the License, or (at your option) any later # version. # # This program is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. See the GNU General Public License for more # details. # # You should have received a copy of the GNU General Public License along with # this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple # Place - Suite 330, Boston, MA 02111-1307, USA. # # from django.shortcuts import render, redirect, get_object_or_404 from django.http import HttpResponseRedirect from django.views.generic import ListView, DetailView, RedirectView 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.forms.models import modelform_factory 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, QuickNotifMixin from core.views.forms import SelectDateTime from core.models import Notification, RealGroup from club.models import Club # Sith object sith = Sith.objects.first class ComTabsMixin(TabedViewMixin): def get_tabs_title(self): return _("Communication administration") def get_list_of_tabs(self): tab_list = [] tab_list.append({ 'url': reverse('com:weekmail'), 'slug': 'weekmail', 'name': _("Weekmail"), }) tab_list.append({ 'url': reverse('com:weekmail_destinations'), 'slug': 'weekmail_destinations', 'name': _("Weekmail destinations"), }) tab_list.append({ 'url': reverse('com:index_edit'), 'slug': 'index', 'name': _("Index page"), }) tab_list.append({ 'url': reverse('com:info_edit'), 'slug': 'info', 'name': _("Info message"), }) tab_list.append({ 'url': reverse('com:alert_edit'), 'slug': 'alert', 'name': _("Alert message"), }) return tab_list class ComEditView(ComTabsMixin, CanEditPropMixin, UpdateView): model = Sith template_name = 'core/edit.jinja' def get_object(self, queryset=None): return Sith.objects.first() class AlertMsgEditView(ComEditView): fields = ['alert_msg'] current_tab = "alert" success_url = reverse_lazy('com:alert_edit') class InfoMsgEditView(ComEditView): fields = ['info_msg'] current_tab = "info" success_url = reverse_lazy('com:info_edit') class IndexEditView(ComEditView): fields = ['index_page'] current_tab = "index" success_url = reverse_lazy('com:index_edit') class WeekmailDestinationEditView(ComEditView): fields = ['weekmail_destinations'] current_tab = "weekmail_destinations" success_url = reverse_lazy('com:weekmail_destinations') # News class NewsForm(forms.ModelForm): class Meta: model = News fields = ['type', 'title', 'club', 'summary', 'content', 'author'] widgets = { 'author': forms.HiddenInput, 'type': forms.RadioSelect, } start_date = forms.DateTimeField(['%Y-%m-%d %H:%M:%S'], label=_("Start date"), widget=SelectDateTime, required=False) end_date = forms.DateTimeField(['%Y-%m-%d %H:%M:%S'], label=_("End date"), widget=SelectDateTime, required=False) until = forms.DateTimeField(['%Y-%m-%d %H:%M:%S'], label=_("Until"), widget=SelectDateTime, required=False) automoderation = forms.BooleanField(label=_("Automoderation"), required=False) def clean(self): self.cleaned_data = super(NewsForm, self).clean() if self.cleaned_data['type'] != "NOTICE": if not self.cleaned_data['start_date']: self.add_error('start_date', ValidationError(_("This field is required."))) if not self.cleaned_data['end_date']: self.add_error('end_date', ValidationError(_("This field is required."))) if self.cleaned_data['start_date'] > self.cleaned_data['end_date']: self.add_error('end_date', ValidationError(_("You crazy? You can not finish an event before starting it."))) if self.cleaned_data['type'] == "WEEKLY" and not self.cleaned_data['until']: self.add_error('until', ValidationError(_("This field is required."))) return self.cleaned_data def save(self): ret = super(NewsForm, self).save() self.instance.dates.all().delete() if self.instance.type == "EVENT" or self.instance.type == "CALL": NewsDate(start_date=self.cleaned_data['start_date'], end_date=self.cleaned_data['end_date'], news=self.instance).save() elif self.instance.type == "WEEKLY": start_date = self.cleaned_data['start_date'] end_date = self.cleaned_data['end_date'] while start_date <= self.cleaned_data['until']: NewsDate(start_date=start_date, end_date=end_date, news=self.instance).save() start_date += timedelta(days=7) end_date += timedelta(days=7) return ret class NewsEditView(CanEditMixin, UpdateView): model = News form_class = NewsForm template_name = 'com/news_edit.jinja' pk_url_kwarg = 'news_id' def get_initial(self): init = {} try: init['start_date'] = self.object.dates.order_by('id').first().start_date.strftime('%Y-%m-%d %H:%M:%S') except: pass try: init['end_date'] = self.object.dates.order_by('id').first().end_date.strftime('%Y-%m-%d %H:%M:%S') except: pass return init def post(self, request, *args, **kwargs): form = self.get_form() if form.is_valid() and not 'preview' in request.POST.keys(): return self.form_valid(form) else: return self.form_invalid(form) def form_valid(self, form): self.object = form.save() if form.cleaned_data['automoderation'] and self.request.user.is_in_group(settings.SITH_GROUP_COM_ADMIN_ID): self.object.moderator = self.request.user self.object.is_moderated = True self.object.save() else: self.object.is_moderated = False self.object.save() for u in RealGroup.objects.filter(id=settings.SITH_GROUP_COM_ADMIN_ID).first().users.all(): if not u.notifications.filter(type="NEWS_MODERATION", viewed=False).exists(): Notification(user=u, url=reverse("com:news_detail", kwargs={'news_id': self.object.id}), type="NEWS_MODERATION").save() return super(NewsEditView, self).form_valid(form) class NewsCreateView(CanCreateMixin, CreateView): model = News form_class = NewsForm template_name = 'com/news_edit.jinja' def get_initial(self): init = {'author': self.request.user} try: init['club'] = Club.objects.filter(id=self.request.GET['club']).first() except: pass return init def post(self, request, *args, **kwargs): form = self.get_form() if form.is_valid() and not 'preview' in request.POST.keys(): return self.form_valid(form) else: self.object = form.instance return self.form_invalid(form) def form_valid(self, form): self.object = form.save() if form.cleaned_data['automoderation'] and self.request.user.is_in_group(settings.SITH_GROUP_COM_ADMIN_ID): self.object.moderator = self.request.user self.object.is_moderated = True self.object.save() else: for u in RealGroup.objects.filter(id=settings.SITH_GROUP_COM_ADMIN_ID).first().users.all(): if not u.notifications.filter(type="NEWS_MODERATION", viewed=False).exists(): Notification(user=u, url=reverse("com:news_detail", kwargs={'news_id': self.object.id}), type="NEWS_MODERATION").save() return super(NewsCreateView, self).form_valid(form) class NewsModerateView(CanEditMixin, SingleObjectMixin): model = News pk_url_kwarg = 'news_id' def get(self, request, *args, **kwargs): self.object = self.get_object() if 'remove' in request.GET.keys(): self.object.is_moderated = False else: self.object.is_moderated = True self.object.moderator = request.user self.object.save() if 'next' in self.request.GET.keys(): return redirect(self.request.GET['next']) return redirect('com:news_admin_list') class NewsAdminListView(CanEditMixin, ListView): model = News template_name = 'com/news_admin_list.jinja' queryset = News.objects.filter(dates__end_date__gte=timezone.now()).distinct().order_by('id') class NewsListView(CanViewMixin, ListView): model = News template_name = 'com/news_list.jinja' def get_context_data(self, **kwargs): kwargs = super(NewsListView, self).get_context_data(**kwargs) kwargs['NewsDate'] = NewsDate kwargs['timedelta'] = timedelta return kwargs class NewsDetailView(CanViewMixin, DetailView): model = News template_name = 'com/news_detail.jinja' pk_url_kwarg = 'news_id' # Weekmail class WeekmailPreviewView(ComTabsMixin, CanEditPropMixin, DetailView): model = Weekmail template_name = 'com/weekmail_preview.jinja' success_url = reverse_lazy('com:weekmail') current_tab = "weekmail" def post(self, request, *args, **kwargs): self.object = self.get_object() try: if request.POST['send'] == "validate": self.object.send() return HttpResponseRedirect(reverse('com:weekmail') + "?qn_weekmail_send_success") except: pass return super(WeekmailEditView, self).get(request, *args, **kwargs) def get_object(self, queryset=None): return self.model.objects.filter(sent=False).order_by('-id').first() def get_context_data(self, **kwargs): """Add rendered weekmail""" kwargs = super(WeekmailPreviewView, self).get_context_data(**kwargs) kwargs['weekmail_rendered'] = self.object.render_html() return kwargs class WeekmailEditView(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView): model = Weekmail template_name = 'com/weekmail.jinja' form_class = modelform_factory(Weekmail, fields=['title', 'intro', 'joke', 'protip', 'conclusion'], help_texts={'title': _("Delete and save to regenerate")}) success_url = reverse_lazy('com:weekmail') current_tab = "weekmail" def get_object(self, queryset=None): weekmail = self.model.objects.filter(sent=False).order_by('-id').first() if not weekmail.title: now = timezone.now() weekmail.title = _("Weekmail of the ") + (now + timedelta(days=6 - now.weekday())).strftime('%d/%m/%Y') weekmail.save() return weekmail def get(self, request, *args, **kwargs): self.object = self.get_object() 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'] 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(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView): """Edit an article""" model = WeekmailArticle fields = ['title', 'club', '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" current_tab = "weekmail" class WeekmailArticleCreateView(QuickNotifMixin, CanViewMixin, CreateView): #XXX need to protect this view """Post an article""" model = WeekmailArticle fields = ['title', 'club', 'content'] template_name = 'core/create.jinja' success_url = reverse_lazy('core:user_tools') quick_notif_url_arg = "qn_weekmail_new_article" def get_initial(self): init = {} try: init['club'] = Club.objects.filter(id=self.request.GET['club']).first() except: pass return init 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(CanEditPropMixin, DeleteView): """Delete an article""" model = WeekmailArticle template_name = 'core/delete_confirm.jinja' success_url = reverse_lazy('com:weekmail') pk_url_kwarg = "article_id"