Sith/core/views/group.py

139 lines
3.8 KiB
Python
Raw Normal View History

#
# Copyright 2023 © AE UTBM
# ae@utbm.fr / ae.info@utbm.fr
#
# This file is part of the website of the UTBM Student Association (AE UTBM),
# https://ae.utbm.fr.
#
# You can find the source code of the website at https://github.com/ae-utbm/sith3
#
# LICENSED UNDER THE GNU GENERAL PUBLIC LICENSE VERSION 3 (GPLv3)
# SEE : https://raw.githubusercontent.com/ae-utbm/sith3/master/LICENSE
# OR WITHIN THE LOCAL FILE "LICENSE"
#
#
2024-07-12 07:34:16 +00:00
"""Views to manage Groups."""
2024-06-24 11:07:36 +00:00
from ajax_select.fields import AutoCompleteSelectMultipleField
from django import forms
from django.urls import reverse_lazy
from django.utils.translation import gettext_lazy as _
2024-06-24 11:07:36 +00:00
from django.views.generic import ListView
from django.views.generic.edit import CreateView, DeleteView, UpdateView
from core.models import RealGroup, User
2021-11-30 16:54:51 +00:00
from core.views import CanCreateMixin, CanEditMixin, DetailFormView
# Forms
class EditMembersForm(forms.Form):
2024-07-12 07:34:16 +00:00
"""Add and remove members from a Group."""
def __init__(self, *args, **kwargs):
self.current_users = kwargs.pop("users", [])
2024-06-27 12:46:43 +00:00
super().__init__(*args, **kwargs)
self.fields["users_removed"] = forms.ModelMultipleChoiceField(
User.objects.filter(id__in=self.current_users).all(),
2019-04-10 23:45:42 +00:00
label=_("Users to remove from group"),
required=False,
widget=forms.CheckboxSelectMultiple,
)
users_added = AutoCompleteSelectMultipleField(
"users",
2019-04-10 23:45:42 +00:00
label=_("Users to add to group"),
help_text=_("Search users to add (one or more)."),
required=False,
)
def clean_users_added(self):
2024-07-12 07:34:16 +00:00
"""Check that the user is not trying to add an user already in the group."""
2024-06-27 12:46:43 +00:00
cleaned_data = super().clean()
users_added = cleaned_data.get("users_added", None)
if not users_added:
return users_added
current_users = [
str(id_) for id_ in self.current_users.values_list("id", flat=True)
]
for user in users_added:
if user in current_users:
raise forms.ValidationError(
_("You can not add the same user twice"), code="invalid"
)
return users_added
# Views
2017-06-12 07:42:03 +00:00
class GroupListView(CanEditMixin, ListView):
2024-07-12 07:34:16 +00:00
"""Displays the Group list."""
2018-10-04 19:29:19 +00:00
2016-03-29 10:45:10 +00:00
model = RealGroup
2018-04-26 16:33:10 +00:00
ordering = ["name"]
2016-02-02 10:00:08 +00:00
template_name = "core/group_list.jinja"
2017-06-12 07:42:03 +00:00
class GroupEditView(CanEditMixin, UpdateView):
2024-07-12 07:34:16 +00:00
"""Edit infos of a Group."""
2016-03-29 10:45:10 +00:00
model = RealGroup
pk_url_kwarg = "group_id"
2016-02-02 10:00:08 +00:00
template_name = "core/group_edit.jinja"
2018-10-04 19:29:19 +00:00
fields = ["name", "description"]
2017-06-12 07:42:03 +00:00
2021-11-30 16:54:51 +00:00
class GroupCreateView(CanCreateMixin, CreateView):
2024-07-12 07:34:16 +00:00
"""Add a new Group."""
2016-05-11 15:54:00 +00:00
model = RealGroup
2021-11-30 16:54:51 +00:00
template_name = "core/create.jinja"
2018-10-04 19:29:19 +00:00
fields = ["name", "description"]
2016-05-11 15:54:00 +00:00
2017-06-12 07:42:03 +00:00
2019-04-10 11:21:34 +00:00
class GroupTemplateView(CanEditMixin, DetailFormView):
2024-07-12 07:34:16 +00:00
"""Display all users in a given Group
Allow adding and removing users from it.
"""
2018-04-26 16:33:10 +00:00
model = RealGroup
form_class = EditMembersForm
2018-04-26 16:33:10 +00:00
pk_url_kwarg = "group_id"
template_name = "core/group_detail.jinja"
def form_valid(self, form):
2024-06-27 12:46:43 +00:00
resp = super().form_valid(form)
data = form.clean()
group = self.get_object()
for user in data["users_removed"]:
group.users.remove(user)
for user in data["users_added"]:
group.users.add(user)
group.save()
return resp
def get_success_url(self):
2019-04-10 11:21:34 +00:00
return reverse_lazy(
"core:group_detail", kwargs={"group_id": self.get_object().id}
)
def get_form_kwargs(self):
2024-06-27 12:46:43 +00:00
kwargs = super().get_form_kwargs()
kwargs["users"] = self.get_object().users.all()
return kwargs
2018-04-26 16:33:10 +00:00
2016-05-11 15:54:00 +00:00
class GroupDeleteView(CanEditMixin, DeleteView):
2024-07-12 07:34:16 +00:00
"""Delete a Group."""
2016-05-11 15:54:00 +00:00
model = RealGroup
pk_url_kwarg = "group_id"
template_name = "core/delete_confirm.jinja"
2018-10-04 19:29:19 +00:00
success_url = reverse_lazy("core:group_list")