improved NoteFormMixin, added Appel model and form in 'suivi'

This commit is contained in:
Arthur Gerbaud
2016-08-17 17:39:24 +02:00
parent d940bc5b1c
commit 9812aa858c
11 changed files with 76 additions and 50 deletions

View File

@@ -3,30 +3,28 @@ from django.shortcuts import redirect
from .forms import *
class SujetNoteFormMixin(object):
class NoteFormMixin(FormMixin):
form_class = AutoNoteForm
form_class = None
def get_form(self, request, **kwargs):
def get_form(self):
# Should add test to ensure this instance class is
# has SingleObjectMixin set with Sujet model ??
kwargs = self.get_form_kwargs()
return self.form_class(
self.request,
sujet=self.get_object()
**kwargs
)
def dispatch(self, request, **kwargs):
self.form = self.get_form(request)
return super().dispatch(request, **kwargs)
def post(self, request, **kwargs):
if self.form.is_valid():
self.form.save()
return redirect(self.get_success_url())
return self.get(request, **kwargs)
form = self.get_form()
if form.is_valid():
form.save()
return self.form_valid(form)
return self.form_invalid(form)
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['note_form'] = self.form
context['note_form'] = self.get_form()
return context