src/internal/views.py
from abc import ABC
from django.conf import settings
from django.contrib import messages
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404
from django.urls import reverse, reverse_lazy
from django.utils import timezone
from django.utils.translation import gettext_lazy as _
from django.views.generic import CreateView, DeleteView, ListView, UpdateView
from django.views.generic.detail import SingleObjectMixin
from contentbox.views import ContentBoxDetailView, ContentBoxUpdateView
from make_queue.models.course import Printer3DCourse
from util.view_utils import CustomFieldsetFormMixin, PreventGetRequestsMixin
from .forms import (
AddMemberForm, ChangeMemberForm, MemberQuitForm, MemberRetireForm, MemberStatusForm, QuoteForm, RestrictedChangeMemberForm, SecretsForm,
SystemAccessValueForm,
)
from .models import Member, Quote, Secret, SystemAccess
class InternalContentBoxDetailView(ContentBoxDetailView):
extra_context = {
'base_template': 'internal/base.html',
}
change_perms = ContentBoxDetailView.change_perms + ('contentbox.change_internal_contentbox',)
class HomeView(InternalContentBoxDetailView):
template_name = 'internal/home.html'
class CommitteeBulletinBoardView(InternalContentBoxDetailView):
template_name = 'internal/committee_bulletin_board.html'
class InternalContentBoxUpdateView(ContentBoxUpdateView):
permission_required = ('contentbox.change_internal_contentbox',)
raise_exception = True
base_template = 'internal/base.html'
def get_form_kwargs(self):
return {
**super().get_form_kwargs(),
'single_language': settings.LANGUAGE_CODE,
}
class MemberListView(ListView):
model = Member
queryset = Member.objects.select_related('user').prefetch_related('committees__group', 'system_accesses')
template_name = 'internal/member_list.html'
context_object_name = 'members'
def get_context_data(self, **kwargs):
context_data = super().get_context_data(**kwargs)
context_data.update({
'StatusAction': MemberStatusForm.StatusAction,
})
if 'pk' in self.kwargs:
context_data.update({
'selected_member': get_object_or_404(Member, pk=self.kwargs['pk']),
})
return context_data
class MemberFormMixin(CustomFieldsetFormMixin, ABC):
model = Member
base_template = 'internal/base.html'
narrow = False
centered_title = False
back_button_text = _("Member list")
class MemberCreateView(PermissionRequiredMixin, MemberFormMixin, CreateView):
permission_required = ('internal.add_member',)
form_class = AddMemberForm
template_name = 'internal/member_create.html'
form_title = _("Add New Member")
back_button_link = reverse_lazy('member_list')
save_button_text = _("Add")
custom_fieldsets = [
{'fields': ('user', 'date_joined'), 'layout_class': "ui two fields"},
{'fields': ('committees',)},
]
def get_success_url(self):
return reverse('member_update', args=(self.object.pk,))
def form_valid(self, form):
user = form.cleaned_data['user']
registration = Printer3DCourse.objects.filter(username=user.username)
if registration.exists():
registration = registration.first()
user.card_number = registration.card_number
registration.user = user
registration.save()
user.save()
return super().form_valid(form)
class MemberUpdateView(PermissionRequiredMixin, MemberFormMixin, UpdateView):
def has_permission(self):
return (self.request.user == self.get_object().user
or self.user_has_edit_perm())
def user_has_edit_perm(self):
return self.request.user.has_perm('internal.can_edit_group_membership')
def get_form_class(self):
if not self.user_has_edit_perm():
return RestrictedChangeMemberForm
return ChangeMemberForm
def get_form_title(self):
full_name = self.get_object().user.get_full_name()
return _("Change Information for {full_name}").format(full_name=full_name)
def get_back_button_link(self):
return self.get_success_url()
def get_custom_fieldsets(self):
full_form = self.user_has_edit_perm()
custom_fieldsets = [
{'fields': ('contact_email', 'phone_number'), 'layout_class': "ui two fields"},
{'fields': ('google_email', 'MAKE_email' if full_form else None), 'layout_class': "ui two fields"},
{'fields': ('study_program', 'ntnu_starting_semester'), 'layout_class': "ui two fields"},
{'fields': ('card_number',), 'layout_class': "ui two fields"},
{'heading': _("Extra information"), 'icon_class': "info circle"},
{'fields': ('github_username', 'discord_username'), 'layout_class': "ui two fields"},
{'fields': ('minecraft_username',), 'layout_class': "ui two fields"},
]
if full_form:
custom_fieldsets.extend([
{'heading': _("Membership information"), 'icon_class': "group"},
{'fields': ('committees', 'role'), 'layout_class': "ui two fields"},
{'fields': ('comment',)},
{'fields': ('guidance_exemption', 'active', 'honorary'), 'layout_class': "ui three fields"},
])
return custom_fieldsets
def get_form_kwargs(self):
kwargs = super().get_form_kwargs()
if self.object.user:
kwargs['initial'].update({
'card_number': self.object.user.card_number,
})
return kwargs
def get_success_url(self):
return self.object.get_absolute_url()
class MemberRetireView(PermissionRequiredMixin, CustomFieldsetFormMixin, UpdateView):
permission_required = ('internal.can_edit_group_membership',)
model = Member
form_class = MemberRetireForm
template_name = 'internal/member_retire.html'
base_template = 'internal/base.html'
narrow = False
centered_title = False
back_button_text = _("Member list")
save_button_text = _("Set retired")
custom_fieldsets = [
{'fields': ('date_quit_or_retired',), 'layout_class': "ui two fields"},
]
def get_form_title(self):
return _("Set member {name} as retired").format(name=self.object.user.get_full_name())
def get_back_button_link(self):
return self.get_success_url()
def get_success_url(self):
return self.object.get_absolute_url()
class MemberQuitView(MemberRetireView):
form_class = MemberQuitForm
template_name = 'internal/member_quit.html'
save_button_text = _("Set quit")
custom_fieldsets = [
{'fields': ('date_quit_or_retired', 'reason_quit'), 'layout_class': "ui two fields"},
]
def get_form_title(self):
return _("Set member {name} as quit").format(name=self.object.user.get_full_name())
class MemberStatusUpdateView(PermissionRequiredMixin, PreventGetRequestsMixin, UpdateView):
permission_required = ('internal.can_edit_group_membership',)
model = Member
form_class = MemberStatusForm
def form_invalid(self, form):
if '__all__' in form.errors:
for error in form.errors['__all__'].data:
if error.code == 'warning_message':
messages.add_message(self.request, messages.WARNING, error.message)
return HttpResponseRedirect(self.get_success_url())
def get_success_url(self):
return self.object.get_absolute_url()
class SystemAccessUpdateView(PermissionRequiredMixin, PreventGetRequestsMixin, UpdateView):
model = SystemAccess
pk_url_kwarg = 'system_access_pk'
form_class = SystemAccessValueForm
def get_queryset(self):
return get_object_or_404(Member, pk=self.kwargs['pk']).system_accesses
def has_permission(self):
system_access: SystemAccess = self.get_object()
return (
system_access.should_be_changed()
and (self.request.user == system_access.member.user
or self.request.user.has_perm('internal.change_systemaccess'))
)
def get_success_url(self):
return self.object.member.get_absolute_url()
class SecretListView(ListView):
model = Secret
template_name = 'internal/secret_list.html'
context_object_name = 'secrets'
extra_context = {
'secrets_shown_seconds': 10,
'secrets_shown_delayed_seconds': 30,
}
def get_queryset(self):
return Secret.objects.visible_to(self.request.user).default_order_by()
class SecretFormMixin(CustomFieldsetFormMixin, ABC):
model = Secret
form_class = SecretsForm
template_name = 'internal/secret_form.html'
success_url = reverse_lazy('secret_list')
base_template = 'internal/base.html'
back_button_link = success_url
back_button_text = _("Secrets list")
class SecretCreateView(PermissionRequiredMixin, SecretFormMixin, CreateView):
permission_required = ('internal.add_secret',)
form_title = _("Add Secret")
save_button_text = _("Add")
class ExistingSecretPermissionRequiredMixin(PermissionRequiredMixin, SingleObjectMixin, ABC):
def get_permission_required(self):
return self.permission_required + self.get_object().extra_view_perms_str_tuple
class SecretUpdateView(ExistingSecretPermissionRequiredMixin, SecretFormMixin, UpdateView):
permission_required = ('internal.change_secret',)
form_title = _("Change Secret")
class SecretDeleteView(ExistingSecretPermissionRequiredMixin, PreventGetRequestsMixin, DeleteView):
permission_required = ('internal.delete_secret',)
model = Secret
success_url = reverse_lazy('secret_list')
class QuoteListView(ListView):
model = Quote
template_name = 'internal/quote_list.html'
context_object_name = 'quotes'
queryset = Quote.objects.order_by('-date').select_related('author')
class QuoteFormMixin(CustomFieldsetFormMixin, ABC):
model = Quote
form_class = QuoteForm
success_url = reverse_lazy('quote_list')
base_template = 'internal/base.html'
back_button_link = success_url
back_button_text = _("Quotes page")
class QuoteCreateView(PermissionRequiredMixin, QuoteFormMixin, CreateView):
permission_required = ('internal.add_quote',)
initial = {
'date': timezone.localdate,
}
form_title = _("Add Quote")
save_button_text = _("Add")
def form_valid(self, form):
form.instance.author = self.request.user
return super().form_valid(form)
class QuoteUpdateView(PermissionRequiredMixin, QuoteFormMixin, UpdateView):
form_title = _("Change Quote")
def has_permission(self):
return (
self.request.user.has_perm('internal.change_quote')
or self.request.user == self.get_object().author
)
class QuoteDeleteView(PermissionRequiredMixin, PreventGetRequestsMixin, DeleteView):
model = Quote
success_url = reverse_lazy('quote_list')
def has_permission(self):
return (
self.request.user.has_perm('internal.delete_quote')
or self.request.user == self.get_object().author
)