122 lines
5.5 KiB
Python
122 lines
5.5 KiB
Python
import nested_admin
|
|
|
|
from django.forms import Form
|
|
from django.db.models.query import QuerySet
|
|
from django.contrib.auth.models import User
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.contrib import admin, messages
|
|
from django.utils.translation import ngettext
|
|
from django.utils.safestring import SafeText
|
|
|
|
from ..models.missions import Training
|
|
from ..models.weekpref import WeekPreference
|
|
|
|
from .training_adm import TrainingInLIne
|
|
from .hourbuilding_adm import HourBuildingInLine
|
|
|
|
from ..custom.colortag import course_color
|
|
from ..actions.exportweek import export_selected
|
|
|
|
from datetime import date
|
|
from typing import Dict, List, Any
|
|
|
|
class WeekPreferenceAdmin(nested_admin.NestedPolymorphicModelAdmin):
|
|
inlines = (TrainingInLIne, HourBuildingInLine, )
|
|
list_display = ("week", "student__surname","student__name", "student__course", "course_color", "student_brief_mix",)
|
|
list_filter = ("week", "student__course",)
|
|
search_fields = ("student__surname","student__name",)
|
|
actions = ("export",)
|
|
|
|
@admin.action(description="Export Selected Preferences")
|
|
def export(self, request: HttpRequest, queryset: QuerySet[WeekPreference]) -> HttpResponse | None:
|
|
if queryset.count() == 0:
|
|
return None
|
|
self.message_user(request, ngettext("Exporting %d row", "Exporting %d rows", queryset.count()) % queryset.count(), messages.SUCCESS)
|
|
return export_selected(request=request, queryset=queryset)
|
|
|
|
@admin.display(description="Mission Count")
|
|
def student_brief_mix(self, obj: WeekPreference) -> SafeText:
|
|
if not obj.student.course:
|
|
return SafeText("")
|
|
return SafeText(f"{Training.objects.filter(weekpref = obj.id).count()}")
|
|
|
|
@admin.display(description="Color")
|
|
def course_color(self, obj: WeekPreference) -> SafeText:
|
|
if not obj.student.course:
|
|
return SafeText("")
|
|
return course_color(obj.student.course.color)
|
|
|
|
# If a user is registered as student hide filters
|
|
def get_list_filter(self, request: HttpRequest) -> List[str]:
|
|
list_filter = super().get_list_filter(request)
|
|
if hasattr(request.user, "student"):
|
|
return []
|
|
return list_filter
|
|
|
|
# If a user is registered as student do not show actions
|
|
def get_actions(self, request: HttpRequest) -> Dict[str, Any]:
|
|
actions = super().get_actions(request)
|
|
if hasattr(request.user, "student"):
|
|
return {}
|
|
return actions
|
|
|
|
# If a user is registered as student show only their preferences
|
|
def get_queryset(self, request: HttpRequest) -> QuerySet[WeekPreference]:
|
|
qs = super().get_queryset(request)
|
|
if hasattr(request.user, "student"):
|
|
return qs.filter(student=request.user.student)
|
|
# If admin show everything
|
|
return qs
|
|
|
|
def get_form(self, request: HttpRequest, obj: WeekPreference | None = None, **kwargs: Dict[str, Any]) -> Form:
|
|
form: Form = super().get_form(request, obj, **kwargs)
|
|
current_week = date.today().isocalendar().week
|
|
|
|
# If form contains the week field
|
|
if "week" in form.base_fields:
|
|
# Set default value as current week
|
|
form.base_fields["week"].initial = current_week
|
|
|
|
# If student is current user making request
|
|
if hasattr(request.user, "student"):
|
|
student = request.user.student
|
|
if "student" in form.base_fields:
|
|
form.base_fields["student"].initial = student
|
|
form.base_fields["student"].disabled = True
|
|
form.base_fields["week"].disabled = True # student cannot change week
|
|
return form
|
|
|
|
# If user is a student deny edit permission for week past the current one
|
|
def has_change_permission(self, request: HttpRequest, obj: WeekPreference | None = None) -> bool:
|
|
if hasattr(request.user, "student") and obj:
|
|
current_week = date.today().isocalendar().week
|
|
if current_week > obj.week:
|
|
return False
|
|
return True
|
|
|
|
# If user is a student deny edit permission for week past the current one
|
|
def has_add_permission(self, request: HttpRequest, obj: WeekPreference | None = None) -> bool:
|
|
return self.has_change_permission(request, obj)
|
|
|
|
# If user is a student deny edit permission for week past the current one
|
|
def has_delete_permission(self, request: HttpRequest, obj: WeekPreference | None = None)-> bool:
|
|
return self.has_change_permission(request, obj)
|
|
|
|
def changeform_view(self, request: HttpRequest, object_id: int | None = None, form_url: str = "", extra_context=None):
|
|
extra_context = extra_context or {}
|
|
if hasattr(request.user, "student") and object_id:
|
|
current_week = date.today().isocalendar().week
|
|
weekpref = WeekPreference.objects.get(id=object_id)
|
|
if current_week > weekpref.week:
|
|
extra_context["show_save"] = False
|
|
extra_context["show_save_and_continue"] = False
|
|
extra_context["show_save_and_add_another"] = False
|
|
extra_context["show_delete"] = False
|
|
return super().changeform_view(request, object_id, form_url, extra_context)
|
|
|
|
def save_model(self, request: HttpRequest, obj, form: Form, change: bool):
|
|
# Imposta automaticamente lo studente se non è già valorizzato
|
|
if hasattr(request.user, "student") and not obj.student_id:
|
|
obj.student = request.user.student
|
|
super().save_model(request, obj, form, change)
|