# # 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.conf import settings from django.contrib.auth.mixins import UserPassesTestMixin from django.core.exceptions import PermissionDenied from django.http import HttpResponse from django.urls import reverse_lazy from django.utils.timezone import localdate from django.views.generic import TemplateView from django.views.generic.edit import FormView from subscription.forms import ( SelectionDateForm, SubscriptionForm, SubscriptionNewUserForm, ) from subscription.models import Subscription class CanCreateSubscriptionMixin(UserPassesTestMixin): def test_func(self): return self.request.user.can_create_subscription class NewSubscription(CanCreateSubscriptionMixin, TemplateView): template_name = "subscription/subscription.jinja" def get_context_data(self, **kwargs): return super().get_context_data(**kwargs) | { "existing_user_form": SubscriptionForm(), "new_user_form": SubscriptionNewUserForm(), } class CreateSubscriptionExistingUserFragment(CanCreateSubscriptionMixin, FormView): """Create a subscription for a user who already exists.""" form_class = SubscriptionForm success_url = reverse_lazy("subscription:subscription-fragment-existing-user") def render_to_response(self, context, **response_kwargs): print(context["form"]) print(str(context["form"])) print(context["form"].render()) return HttpResponse(context["form"].render(), **response_kwargs) # def dispatch(self, request, *args, **kwargs): # s = SubscriptionForm() # print(s) # return super().dispatch(request, *args, **kwargs) # def form_valid(self, form): # duration = settings.SITH_SUBSCRIPTIONS[form.instance.subscription_type][ # "duration" # ] # form.instance.subscription_start = Subscription.compute_start( # duration=duration, user=form.instance.member # ) # form.instance.subscription_end = Subscription.compute_end( # duration=duration, # start=form.instance.subscription_start, # user=form.instance.member, # ) # return super().form_valid(form) class SubscriptionsStatsView(FormView): template_name = "subscription/stats.jinja" form_class = SelectionDateForm success_url = reverse_lazy("subscriptions:stats") def dispatch(self, request, *arg, **kwargs): self.start_date = localdate() self.end_date = self.start_date if request.user.is_root or request.user.is_board_member: return super().dispatch(request, *arg, **kwargs) raise PermissionDenied def post(self, request, *args, **kwargs): self.form = self.get_form() self.start_date = self.form["start_date"] self.end_date = self.form["end_date"] return super().post(request, *args, **kwargs) def get_initial(self): return { "start_date": self.start_date.strftime("%Y-%m-%d %H:%M:%S"), "end_date": self.end_date.strftime("%Y-%m-%d %H:%M:%S"), } def get_context_data(self, **kwargs): kwargs = super().get_context_data(**kwargs) kwargs["subscriptions_total"] = Subscription.objects.filter( subscription_end__gte=self.end_date, subscription_start__lte=self.start_date ) kwargs["subscriptions_types"] = settings.SITH_SUBSCRIPTIONS kwargs["payment_types"] = settings.SITH_COUNTER_PAYMENT_METHOD kwargs["locations"] = settings.SITH_SUBSCRIPTION_LOCATIONS return kwargs