120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
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
|
|
message_template = 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
|
|
context['message_template'] = self.message_template
|
|
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'
|
|
message_template = 'manuels/supplies_message.html'
|
|
|
|
|
|
def clear_teacher_view(request):
|
|
if request.session['teacher_pk']:
|
|
del request.session['teacher_pk']
|
|
return redirect('home_page')
|