# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay # SPDX-License-Identifier: GPL-3.0-or-later from datetime import datetime from django.contrib.auth.mixins import LoginRequiredMixin from django.contrib.contenttypes.models import ContentType from django.db.models import F, Q from django.urls import reverse_lazy from django.views.generic import CreateView, DetailView, UpdateView, TemplateView from django.utils.translation import gettext_lazy as _ from django_tables2.views import SingleTableView from note.models import NoteUser, Alias, NoteSpecial from permission.backends import PermissionBackend from .forms import ActivityForm, GuestForm from .models import Activity, Guest, Entry from .tables import ActivityTable, GuestTable, EntryTable class ActivityCreateView(LoginRequiredMixin, CreateView): model = Activity form_class = ActivityForm def get_success_url(self, **kwargs): return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]}) class ActivityListView(LoginRequiredMixin, SingleTableView): model = Activity table_class = ActivityTable def get_context_data(self, **kwargs): ctx = super().get_context_data(**kwargs) ctx['title'] = _("Activities") upcoming_activities = Activity.objects.filter(date_end__gt=datetime.now()) ctx['upcoming'] = ActivityTable(data=upcoming_activities) return ctx class ActivityDetailView(LoginRequiredMixin, DetailView): model = Activity context_object_name = "activity" def get_context_data(self, **kwargs): ctx = super().get_context_data() table = GuestTable(data=Guest.objects.filter(activity=self.object) .filter(PermissionBackend.filter_queryset(self.request.user, Guest, "view"))) ctx["guests"] = table return ctx class ActivityUpdateView(LoginRequiredMixin, UpdateView): model = Activity form_class = ActivityForm def get_success_url(self, **kwargs): return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]}) class ActivityInviteView(LoginRequiredMixin, CreateView): model = Guest form_class = GuestForm template_name = "activity/activity_invite.html" def form_valid(self, form): form.instance.activity = Activity.objects.get(pk=self.kwargs["pk"]) return super().form_valid(form) def get_success_url(self, **kwargs): return reverse_lazy('activity:activity_detail', kwargs={"pk": self.kwargs["pk"]}) class ActivityEntryView(LoginRequiredMixin, TemplateView): template_name = "activity/activity_entry.html" def get_context_data(self, **kwargs): ctx = super().get_context_data(**kwargs) activity = Activity.objects.get(pk=self.kwargs["pk"]) ctx["activity"] = activity matched = [] pattern = "^$" if "search" in self.request.GET: pattern = self.request.GET["search"] if not pattern: pattern = "^$" if pattern[0] != "^": pattern = "^" + pattern 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)))\ .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) | Q(name__regex=pattern) | Q(normalized_name__regex=Alias.normalize(pattern))))\ .distinct("username")[:20] for note in note_qs: note.type = "Adhérent" note.activity = activity matched.append(note) table = EntryTable(data=matched) ctx["table"] = table ctx["entries"] = Entry.objects.filter(activity=activity) ctx["title"] = _('Entry for activity "{}"').format(activity.name) ctx["noteuser_ctype"] = ContentType.objects.get_for_model(NoteUser).pk ctx["notespecial_ctype"] = ContentType.objects.get_for_model(NoteSpecial).pk return ctx