From eaeb28a650ea23150f8088e28c0cab422b65f438 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Fri, 30 Jan 2015 13:37:00 +0100 Subject: [PATCH] add script period detection --- teleforma/exam/views.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/teleforma/exam/views.py b/teleforma/exam/views.py index 2154d9e0..765a5f45 100644 --- a/teleforma/exam/views.py +++ b/teleforma/exam/views.py @@ -15,7 +15,10 @@ class ScriptView(CourseAccessMixin, UpdateView): model = Script template_name='exam/script_detail.html' form_class = ScriptForm - success_url = reverse_lazy('teleforma-exam-scripts-pending', kwargs={'period_id':1}) + + def get_success_url(self): + period = Period.objects.get(id=self.kwargs['period_id']) + return reverse_lazy('teleforma-exam-scripts-pending', kwargs={'period_id':period.id}) def get_context_data(self, **kwargs): context = super(ScriptView, self).get_context_data(**kwargs) @@ -123,7 +126,10 @@ class ScriptCreateView(CreateView): model = Script template_name='exam/script_form.html' form_class = ScriptForm - success_url = reverse_lazy('teleforma-exam-scripts-pending', kwargs={'period_id':1}) + + def get_success_url(self): + period = Period.objects.get(id=self.kwargs['period_id']) + return reverse_lazy('teleforma-exam-scripts-pending', kwargs={'period_id':period.id}) def form_valid(self, form): form.instance.author = self.request.user -- 2.39.5