views.py 14 KB
Newer Older
Skia's avatar
Skia committed
1
from django.shortcuts import render, redirect, get_object_or_404
Skia's avatar
Skia committed
2
from django.http import HttpResponseRedirect
3
from django.views.generic import ListView, DetailView, RedirectView
Skia's avatar
Skia committed
4
from django.views.generic.edit import UpdateView, CreateView, DeleteView
Skia's avatar
Skia committed
5
from django.views.generic.detail import SingleObjectMixin
Skia's avatar
Skia committed
6
from django.utils.translation import ugettext_lazy as _
7
from django.core.urlresolvers import reverse, reverse_lazy
8 9
from django.core.exceptions import ValidationError
from django.utils import timezone
Skia's avatar
Skia committed
10
from django.conf import settings
Skia's avatar
Skia committed
11
from django.db.models import Max
Skia's avatar
Skia committed
12
from django.forms.models import modelform_factory
13
from django import forms
14

15 16
from datetime import timedelta

Skia's avatar
Skia committed
17
from com.models import Sith, News, NewsDate, Weekmail, WeekmailArticle
Skia's avatar
Skia committed
18
from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin, TabedViewMixin, CanCreateMixin, QuickNotifMixin
19
from core.views.forms import SelectDateTime
Skia's avatar
Skia committed
20
from core.models import Notification, RealGroup
21 22
from club.models import Club

23

24
# Sith object
25 26 27 28 29 30 31 32 33

sith = Sith.objects.first

class ComTabsMixin(TabedViewMixin):
    def get_tabs_title(self):
        return _("Communication administration")

    def get_list_of_tabs(self):
        tab_list = []
Skia's avatar
Skia committed
34 35 36 37 38
        tab_list.append({
                    'url': reverse('com:weekmail'),
                    'slug': 'weekmail',
                    'name': _("Weekmail"),
                    })
Skia's avatar
Skia committed
39 40 41 42 43
        tab_list.append({
                    'url': reverse('com:weekmail_destinations'),
                    'slug': 'weekmail_destinations',
                    'name': _("Weekmail destinations"),
                    })
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
        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')
82

Skia's avatar
Skia committed
83 84 85 86 87
class WeekmailDestinationEditView(ComEditView):
    fields = ['weekmail_destinations']
    current_tab = "weekmail_destinations"
    success_url = reverse_lazy('com:weekmail_destinations')

88 89 90 91 92
# News

class NewsForm(forms.ModelForm):
    class Meta:
        model = News
Skia's avatar
Skia committed
93
        fields = ['type', 'title', 'club', 'summary', 'content', 'author']
94
        widgets = {
Skia's avatar
Skia committed
95
                'author': forms.HiddenInput,
96 97 98 99 100
                '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)
Skia's avatar
Skia committed
101
    automoderation = forms.BooleanField(label=_("Automoderation"), required=False)
102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131

    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

Skia's avatar
Skia committed
132
class NewsEditView(CanEditMixin, UpdateView):
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
    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

Skia's avatar
Skia committed
148 149 150 151 152 153 154
    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)

Skia's avatar
Skia committed
155 156 157 158 159 160 161 162 163
    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()
Skia's avatar
Skia committed
164 165 166
            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()
Skia's avatar
Skia committed
167 168
        return super(NewsEditView, self).form_valid(form)

Skia's avatar
Skia committed
169
class NewsCreateView(CanCreateMixin, CreateView): #XXX no can_be_created_by function in News model
170 171 172 173 174
    model = News
    form_class = NewsForm
    template_name = 'com/news_edit.jinja'

    def get_initial(self):
Skia's avatar
Skia committed
175
        init = {'author': self.request.user}
176 177 178 179 180
        try:
            init['club'] = Club.objects.filter(id=self.request.GET['club']).first()
        except: pass
        return init

Skia's avatar
Skia committed
181 182 183 184 185 186 187 188
    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)

Skia's avatar
Skia committed
189 190 191 192 193 194
    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()
Skia's avatar
Skia committed
195 196 197 198
        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()
Skia's avatar
Skia committed
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
        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):
215 216
    model = News
    template_name = 'com/news_admin_list.jinja'
Skia's avatar
Skia committed
217
    queryset = News.objects.filter(dates__end_date__gte=timezone.now()).distinct().order_by('id')
218

Skia's avatar
Skia committed
219
class NewsListView(CanViewMixin, ListView):
220 221 222
    model = News
    template_name = 'com/news_list.jinja'

Skia's avatar
Skia committed
223 224 225 226 227 228 229
    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):
230 231 232
    model = News
    template_name = 'com/news_detail.jinja'
    pk_url_kwarg = 'news_id'
Skia's avatar
Skia committed
233

Skia's avatar
Skia committed
234 235
# Weekmail

Skia's avatar
Skia committed
236
class WeekmailPreviewView(ComTabsMixin, CanEditPropMixin, DetailView):
Skia's avatar
Skia committed
237 238 239
    model = Weekmail
    template_name = 'com/weekmail_preview.jinja'
    success_url = reverse_lazy('com:weekmail')
Skia's avatar
Skia committed
240 241 242 243 244 245 246 247 248 249
    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)
Skia's avatar
Skia committed
250 251 252 253 254 255 256

    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)
Skia's avatar
Skia committed
257
        kwargs['weekmail_rendered'] = self.object.render_html()
Skia's avatar
Skia committed
258 259
        return kwargs

Skia's avatar
Skia committed
260
class WeekmailEditView(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView):
Skia's avatar
Skia committed
261 262
    model = Weekmail
    template_name = 'com/weekmail.jinja'
Skia's avatar
Skia committed
263 264
    form_class = modelform_factory(Weekmail, fields=['title', 'intro', 'joke', 'protip', 'conclusion'],
            help_texts={'title': _("Delete and save to regenerate")})
Skia's avatar
Skia committed
265
    success_url = reverse_lazy('com:weekmail')
Skia's avatar
Skia committed
266
    current_tab = "weekmail"
Skia's avatar
Skia committed
267 268

    def get_object(self, queryset=None):
Skia's avatar
Skia committed
269 270 271 272
        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')
Skia's avatar
Skia committed
273
            weekmail.save()
Skia's avatar
Skia committed
274
        return weekmail
Skia's avatar
Skia committed
275

Skia's avatar
Skia committed
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
    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

Skia's avatar
Skia committed
315
class WeekmailArticleEditView(ComTabsMixin, QuickNotifMixin, CanEditPropMixin, UpdateView):
Skia's avatar
Skia committed
316 317
    """Edit an article"""
    model = WeekmailArticle
318
    fields = ['title', 'club', 'content']
Skia's avatar
Skia committed
319 320 321 322
    pk_url_kwarg = "article_id"
    template_name = 'core/edit.jinja'
    success_url = reverse_lazy('com:weekmail')
    quick_notif_url_arg = "qn_weekmail_article_edit"
Skia's avatar
Skia committed
323
    current_tab = "weekmail"
Skia's avatar
Skia committed
324

Skia's avatar
Skia committed
325
class WeekmailArticleCreateView(QuickNotifMixin, CreateView): #XXX need to protect this view
Skia's avatar
Skia committed
326 327
    """Post an article"""
    model = WeekmailArticle
328
    fields = ['title', 'club', 'content']
Skia's avatar
Skia committed
329 330 331 332
    template_name = 'core/create.jinja'
    success_url = reverse_lazy('core:user_tools')
    quick_notif_url_arg = "qn_weekmail_new_article"

333 334 335
    def get_initial(self):
        init = {}
        try:
Skia's avatar
Skia committed
336
            init['club'] = Club.objects.filter(id=self.request.GET['club']).first()
337 338 339
        except: pass
        return init

Skia's avatar
Skia committed
340
    def form_valid(self, form):
341 342
        # club = get_object_or_404(Club, id=self.kwargs['club_id'])
        # form.instance.club = club
Skia's avatar
Skia committed
343 344 345
        form.instance.author = self.request.user
        return super(WeekmailArticleCreateView, self).form_valid(form)

Skia's avatar
Skia committed
346
class WeekmailArticleDeleteView(CanEditPropMixin, DeleteView):
Skia's avatar
Skia committed
347 348 349
    """Delete an article"""
    model = WeekmailArticle
    template_name = 'core/delete_confirm.jinja'
350
    pk_url_kwarg = "article_id"
Skia's avatar
Skia committed
351 352 353 354




Skia's avatar
Skia committed
355