Sith/counter/views/student_card.py

#
# 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/sith
#
# LICENSED UNDER THE GNU GENERAL PUBLIC LICENSE VERSION 3 (GPLv3)
# SEE : https://raw.githubusercontent.com/ae-utbm/sith/master/LICENSE
# OR WITHIN THE LOCAL FILE "LICENSE"
#
#


from django.core.exceptions import PermissionDenied
from django.http import Http404, HttpRequest, HttpResponse
from django.shortcuts import get_object_or_404
from django.urls import reverse
from django.utils.translation import gettext as _
from django.views.generic.edit import DeleteView, FormView

from core.utils import FormFragmentTemplateData
from core.views import can_edit
from counter.forms import StudentCardForm
from counter.models import Customer, StudentCard
from counter.utils import is_logged_in_counter


class StudentCardDeleteView(DeleteView):
    """View used to delete a card from a user. This is a fragment view !"""

    model = StudentCard
    template_name = "counter/fragments/delete_student_card.jinja"

    def dispatch(self, request: HttpRequest, *args, **kwargs):
        self.customer = get_object_or_404(Customer, pk=kwargs["customer_id"])
        if not is_logged_in_counter(request) and not can_edit(
            self.get_object(), request.user
        ):
            raise PermissionDenied()
        return super().dispatch(request, *args, **kwargs)

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["action"] = self.request.path
        context["action_cancel"] = self.get_success_url()
        return context

    def get_object(self, queryset=None):
        if not hasattr(self.customer, "student_card"):
            raise Http404(
                _("%(name)s has no registered student card")
                % {"name": self.customer.user.get_full_name()}
            )
        return self.customer.student_card

    def get_success_url(self, **kwargs):
        return reverse(
            "counter:add_student_card", kwargs={"customer_id": self.customer.pk}
        )


class StudentCardFormView(FormView):
    """Add a new student card. This is a fragment view !"""

    form_class = StudentCardForm
    template_name = "counter/fragments/create_student_card.jinja"

    @classmethod
    def get_template_data(
        cls, customer: Customer
    ) -> FormFragmentTemplateData[StudentCardForm]:
        """Get necessary data to pre-render the fragment"""
        return FormFragmentTemplateData(
            form=cls.form_class(),
            template=cls.template_name,
            context={
                "action": reverse(
                    "counter:add_student_card", kwargs={"customer_id": customer.pk}
                ),
                "customer": customer,
            },
        )

    def dispatch(self, request: HttpRequest, *args, **kwargs):
        self.customer = get_object_or_404(
            Customer.objects.select_related("student_card"), pk=kwargs["customer_id"]
        )

        if not is_logged_in_counter(request) and not StudentCard.can_create(
            self.customer, request.user
        ):
            raise PermissionDenied

        return super().dispatch(request, *args, **kwargs)

    def form_valid(self, form: StudentCardForm) -> HttpResponse:
        data = form.clean()
        StudentCard.objects.update_or_create(
            customer=self.customer, defaults={"uid": data["uid"]}
        )
        return super().form_valid(form)

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        data = self.get_template_data(self.customer)
        context.update(data.context)
        return context

    def get_success_url(self, **kwargs):
        return self.request.path