nk20/apps/activity/views.py

161 lines
6.3 KiB
Python
Raw Normal View History

2020-03-01 16:16:38 +00:00
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later
from datetime import datetime, timezone
2020-03-01 16:16:38 +00:00
from django.contrib.auth.mixins import LoginRequiredMixin
2020-03-28 00:45:13 +00:00
from django.contrib.contenttypes.models import ContentType
from django.db.models import F, Q
from django.urls import reverse_lazy
2020-03-26 23:40:35 +00:00
from django.views.generic import CreateView, DetailView, UpdateView, TemplateView
2020-03-27 00:31:54 +00:00
from django.utils.translation import gettext_lazy as _
2020-03-01 16:16:38 +00:00
from django_tables2.views import SingleTableView
from note.models import NoteUser, Alias, NoteSpecial
2020-03-27 20:18:27 +00:00
from permission.backends import PermissionBackend
from permission.views import ProtectQuerysetMixin
2020-03-28 00:45:13 +00:00
2020-03-27 17:02:22 +00:00
from .forms import ActivityForm, GuestForm
from .models import Activity, Guest, Entry
2020-03-28 00:45:13 +00:00
from .tables import ActivityTable, GuestTable, EntryTable
2020-03-26 23:40:35 +00:00
2020-03-01 16:16:38 +00:00
class ActivityCreateView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
2020-03-01 16:16:38 +00:00
model = Activity
2020-03-27 00:31:54 +00:00
form_class = ActivityForm
2020-03-27 20:18:27 +00:00
def form_valid(self, form):
form.instance.creater = self.request.user
return super().form_valid(form)
2020-03-27 20:18:27 +00:00
def get_success_url(self, **kwargs):
self.object.refresh_from_db()
return reverse_lazy('activity:activity_detail', kwargs={"pk": self.object.pk})
2020-03-26 23:40:35 +00:00
class ActivityListView(ProtectQuerysetMixin, LoginRequiredMixin, SingleTableView):
2020-03-01 16:16:38 +00:00
model = Activity
2020-03-27 17:02:22 +00:00
table_class = ActivityTable
2020-03-27 00:31:54 +00:00
def get_queryset(self):
return super().get_queryset().reverse()
2020-03-27 00:31:54 +00:00
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
2020-03-27 00:31:54 +00:00
context['title'] = _("Activities")
upcoming_activities = Activity.objects.filter(date_end__gt=datetime.now())
context['upcoming'] = ActivityTable(data=upcoming_activities
.filter(PermissionBackend.filter_queryset(self.request.user, Activity, "view")))
2020-03-27 00:31:54 +00:00
return context
2020-03-01 16:16:38 +00:00
2020-03-26 23:40:35 +00:00
class ActivityDetailView(ProtectQuerysetMixin, LoginRequiredMixin, DetailView):
2020-03-01 16:16:38 +00:00
model = Activity
2020-03-27 18:47:43 +00:00
context_object_name = "activity"
2020-03-01 16:16:38 +00:00
2020-03-27 20:18:27 +00:00
def get_context_data(self, **kwargs):
context = super().get_context_data()
2020-03-27 20:18:27 +00:00
table = GuestTable(data=Guest.objects.filter(activity=self.object)
.filter(PermissionBackend.filter_queryset(self.request.user, Guest, "view")))
context["guests"] = table
2020-03-27 20:18:27 +00:00
context["activity_started"] = datetime.now(timezone.utc) > self.object.date_start
return context
2020-03-27 20:18:27 +00:00
2020-03-26 23:40:35 +00:00
class ActivityUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
2020-03-01 16:16:38 +00:00
model = Activity
2020-03-27 00:31:54 +00:00
form_class = ActivityForm
2020-03-27 20:18:27 +00:00
def get_success_url(self, **kwargs):
return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]})
2020-03-26 23:40:35 +00:00
class ActivityInviteView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
2020-03-27 17:02:22 +00:00
model = Guest
form_class = GuestForm
template_name = "activity/activity_invite.html"
def get_form(self, form_class=None):
form = super().get_form(form_class)
form.activity = Activity.objects.filter(PermissionBackend.filter_queryset(self.request.user, Activity, "view"))\
.get(pk=self.kwargs["pk"])
return form
2020-03-27 18:16:38 +00:00
def form_valid(self, form):
form.instance.activity = Activity.objects\
.filter(PermissionBackend.filter_queryset(self.request.user, Activity, "view")).get(pk=self.kwargs["pk"])
2020-03-27 18:16:38 +00:00
return super().form_valid(form)
def get_success_url(self, **kwargs):
return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]})
2020-03-27 17:02:22 +00:00
class ActivityEntryView(LoginRequiredMixin, TemplateView):
2020-03-28 00:45:13 +00:00
template_name = "activity/activity_entry.html"
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
2020-03-28 00:45:13 +00:00
activity = Activity.objects.filter(PermissionBackend.filter_queryset(self.request.user, Activity, "view"))\
.get(pk=self.kwargs["pk"])
context["activity"] = activity
2020-03-28 00:45:13 +00:00
matched = []
pattern = "^$"
if "search" in self.request.GET:
pattern = self.request.GET["search"]
2020-03-28 12:38:31 +00:00
if not pattern:
pattern = "^$"
if pattern[0] != "^":
pattern = "^" + pattern
2020-03-28 00:45:13 +00:00
guest_qs = Guest.objects\
.annotate(balance=F("inviter__balance"), note_name=F("inviter__user__username"))\
.filter(Q(first_name__regex=pattern) | Q(last_name__regex=pattern)
| Q(inviter__alias__name__regex=pattern)
| Q(inviter__alias__normalized_name__regex=Alias.normalize(pattern))) \
.filter(PermissionBackend.filter_queryset(self.request.user, Guest, "view"))\
2020-03-28 00:45:13 +00:00
.distinct()[:20]
for guest in guest_qs:
guest.type = "Invité"
matched.append(guest)
note_qs = Alias.objects.annotate(last_name=F("note__noteuser__user__last_name"),
first_name=F("note__noteuser__user__first_name"),
username=F("note__noteuser__user__username"),
note_name=F("name"),
balance=F("note__balance"))\
.filter(Q(note__polymorphic_ctype__model="noteuser")
& (Q(note__noteuser__user__first_name__regex=pattern)
| Q(note__noteuser__user__last_name__regex=pattern)
2020-03-28 12:38:31 +00:00
| Q(name__regex=pattern)
| Q(normalized_name__regex=Alias.normalize(pattern)))) \
.filter(PermissionBackend.filter_queryset(self.request.user, Alias, "view"))\
.distinct()[:20]
2020-03-28 00:45:13 +00:00
for note in note_qs:
note.type = "Adhérent"
note.activity = activity
2020-03-28 00:45:13 +00:00
matched.append(note)
table = EntryTable(data=matched)
context["table"] = table
2020-03-28 00:45:13 +00:00
context["entries"] = Entry.objects.filter(activity=activity)
context["title"] = _('Entry for activity "{}"').format(activity.name)
context["noteuser_ctype"] = ContentType.objects.get_for_model(NoteUser).pk
context["notespecial_ctype"] = ContentType.objects.get_for_model(NoteSpecial).pk
context["activities_open"] = Activity.objects.filter(open=True).filter(
PermissionBackend.filter_queryset(self.request.user, Activity, "view")).filter(
PermissionBackend.filter_queryset(self.request.user, Activity, "change")).all()
return context