from django.contrib import messages from django.shortcuts import get_object_or_404, redirect from django.urls import reverse from django.views.generic import CreateView, ListView from manuels.forms import AddBookForm, AddSuppliesForm from manuels.models import Teacher, Book, SuppliesRequirement import logging logger = logging.getLogger(__name__) class HomePageView(CreateView): model = Teacher fields = '__all__' template_name = 'manuels/home_page.html' def get(self, request, *args, **kwargs): teacher_pk = request.session.get('teacher_pk') if teacher_pk: return redirect('list_books', pk=teacher_pk) return super().get(request, *args, **kwargs) def form_valid(self, form): response = super().form_valid(form) self.object.send_link(self.request) return response class BaseTeacherView: teacher = None def dispatch(self, request, *args, **kwargs): self.teacher = get_object_or_404(Teacher, pk=self.kwargs['pk']) request.session['teacher_pk'] = str(self.teacher.pk) return super().dispatch(request, *args, **kwargs) def get_context_data(self, **kwargs): context = super().get_context_data() context['teacher'] = self.teacher return context class ListBooksView(BaseTeacherView, ListView): model = Book template_name = 'manuels/list_books_supplies.html' context_object_name = 'books' def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['supplies'] = SuppliesRequirement.objects.filter(teacher=self.teacher) return context def get_queryset(self): return Book.objects.filter(teacher=self.teacher) class AddItemView(BaseTeacherView, CreateView): add_another = False item_text = None item_text_plural = None success_target = None def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['item'] = self.item_text context['item_plural'] = self.item_text_plural return context def get_initial(self): return { 'teacher': self.teacher } def get_form(self, form_class=None): form = super().get_form(form_class) form.fields['teacher'].queryset = Teacher.objects.filter(pk=self.teacher.pk) return form def form_valid(self, form): self.add_another = form.cleaned_data['add_another'] return super().form_valid(form) def get_success_url(self): messages.success(self.request, f'"{self.object}" a été ajouté.') if self.add_another: return reverse(self.success_target, args=[str(self.teacher.pk)]) else: return reverse('list_books', args=[str(self.teacher.pk)]) class AddBookView(AddItemView): model = Book template_name = 'manuels/add_item.html' form_class = AddBookForm success_target = 'add_book' item_text = 'un livre' item_text_plural = 'livres' class AddSuppliesView(AddItemView): model = SuppliesRequirement template_name = 'manuels/add_item.html' form_class = AddSuppliesForm success_target = 'add_supplies' item_text = 'des fournitures' item_text_plural = 'fournitures' def clear_teacher_view(request): if request.session['teacher_pk']: del request.session['teacher_pk'] return redirect('home_page')