Sith/com/views.py

346 lines
14 KiB
Python
Raw Normal View History

2017-01-07 14:08:23 +00:00
from django.shortcuts import render, redirect, get_object_or_404
2017-01-11 11:18:42 +00:00
from django.http import HttpResponseRedirect
from django.views.generic import ListView, DetailView, RedirectView
2017-01-07 14:08:23 +00:00
from django.views.generic.edit import UpdateView, CreateView, DeleteView
2016-12-23 17:40:12 +00:00
from django.views.generic.detail import SingleObjectMixin
2016-12-23 02:17:38 +00:00
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
2016-12-23 19:11:27 +00:00
from django.conf import settings
2017-01-07 14:08:23 +00:00
from django.db.models import Max
from django import forms
from datetime import timedelta
2017-01-02 23:07:45 +00:00
from com.models import Sith, News, NewsDate, Weekmail, WeekmailArticle
2017-01-07 14:08:23 +00:00
from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, TabedViewMixin, CanCreateMixin, QuickNotifMixin
from core.views.forms import SelectDateTime
2016-12-23 19:11:27 +00:00
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 = []
2017-01-02 23:07:45 +00:00
tab_list.append({
'url': reverse('com:weekmail'),
'slug': 'weekmail',
'name': _("Weekmail"),
})
2017-01-11 00:34:16 +00:00
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')
2017-01-11 00:34:16 +00:00
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
2016-12-23 17:40:12 +00:00
fields = ['type', 'title', 'club', 'summary', 'content', 'author']
widgets = {
2016-12-23 17:40:12 +00:00
'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)
2016-12-23 17:40:12 +00:00
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['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
2016-12-23 17:40:12 +00:00
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
2016-12-23 18:43:59 +00:00
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)
2016-12-23 17:40:12 +00:00
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()
2016-12-23 19:11:27 +00:00
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()
2016-12-23 17:40:12 +00:00
return super(NewsEditView, self).form_valid(form)
2017-01-15 22:09:30 +00:00
class NewsCreateView(CanCreateMixin, CreateView): #XXX no can_be_created_by function in News model
model = News
form_class = NewsForm
template_name = 'com/news_edit.jinja'
def get_initial(self):
2016-12-23 17:40:12 +00:00
init = {'author': self.request.user}
try:
init['club'] = Club.objects.filter(id=self.request.GET['club']).first()
except: pass
return init
2016-12-23 18:43:59 +00:00
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)
2016-12-23 17:40:12 +00:00
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()
2016-12-23 19:11:27 +00:00
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()
2016-12-23 17:40:12 +00:00
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()
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'
2016-12-23 17:40:12 +00:00
queryset = News.objects.filter(dates__end_date__gte=timezone.now()).distinct().order_by('id')
2016-12-23 17:40:12 +00:00
class NewsListView(CanViewMixin, ListView):
model = News
template_name = 'com/news_list.jinja'
2016-12-23 17:40:12 +00:00
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'
2016-12-23 17:40:12 +00:00
2017-01-02 23:07:45 +00:00
# Weekmail
2017-01-15 22:09:30 +00:00
class WeekmailPreviewView(ComTabsMixin, CanEditPropMixin, DetailView):
2017-01-10 17:30:27 +00:00
model = Weekmail
template_name = 'com/weekmail_preview.jinja'
success_url = reverse_lazy('com:weekmail')
2017-01-11 11:18:42 +00:00
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)
2017-01-10 17:30:27 +00:00
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)
2017-01-11 00:34:16 +00:00
kwargs['weekmail_rendered'] = self.object.render_html()
2017-01-10 17:30:27 +00:00
return kwargs
2017-01-15 22:09:30 +00:00
class WeekmailEditView(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView):
2017-01-02 23:07:45 +00:00
model = Weekmail
template_name = 'com/weekmail.jinja'
2017-01-03 15:50:53 +00:00
fields = ['title', 'intro', 'joke', 'protip', 'conclusion']
success_url = reverse_lazy('com:weekmail')
2017-01-11 00:34:16 +00:00
current_tab = "weekmail"
2017-01-02 23:07:45 +00:00
def get_object(self, queryset=None):
2017-01-03 15:50:53 +00:00
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')
2017-01-10 17:30:27 +00:00
weekmail.save()
2017-01-03 15:50:53 +00:00
return weekmail
2017-01-02 23:07:45 +00:00
2017-01-07 14:08:23 +00:00
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
2017-01-15 22:09:30 +00:00
class WeekmailArticleEditView(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView):
2017-01-07 14:08:23 +00:00
"""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"
2017-01-11 00:34:16 +00:00
current_tab = "weekmail"
2017-01-07 14:08:23 +00:00
2017-01-15 22:09:30 +00:00
class WeekmailArticleCreateView(QuickNotifMixin, CreateView): #XXX need to protect this view
2017-01-07 14:08:23 +00:00
"""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)
2017-01-15 22:09:30 +00:00
class WeekmailArticleDeleteView(CanEditPropMixin, DeleteView):
2017-01-07 14:08:23 +00:00
"""Delete an article"""
model = WeekmailArticle
template_name = 'core/delete_confirm.jinja'
2016-12-23 17:40:12 +00:00