From ef973149b6650d970ba0a11b120f53a7f4ad07b6 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Wed, 3 Feb 2016 23:48:15 +0100 Subject: [PATCH] fix context --- teleforma/views/crfpa.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/teleforma/views/crfpa.py b/teleforma/views/crfpa.py index a2b25f3e..65cee9b4 100644 --- a/teleforma/views/crfpa.py +++ b/teleforma/views/crfpa.py @@ -425,17 +425,18 @@ class UserCompleteView(TemplateView): class RegistrationPDFView(PDFTemplateResponseMixin, DetailView): - context_object_name = "student" - model = Student + model = User template_name = 'teleforma/registration_pdf.html' pdf_template_name = template_name def get_context_data(self, **kwargs): - student = self.get_object() + context = super RegistrationPDFView, self).get_context_data(**kwargs) + user = User.objects.get(username=kwargs['username']) + student = user.student.all()[0] if student.training and not student.trainings: student.trainings.add(student.training) student.save() - context = super RegistrationPDFView, self).get_context_data(**kwargs) + context['student'] = student return context @@ -445,8 +446,8 @@ class RegistrationPDFViewDowload(RegistrationPDFView): def get_pdf_filename(self): super(RegistrationPDFViewDowload, self).get_pdf_filename() - student = self.get_object() + student = self.context['student'] prefix = unicode(_('Registration')) - filename = '_'.join([prefix, student.first_name, student.user.last_name]) + filename = '_'.join([prefix, student.user.first_name, student.user.last_name]) filename += '.pdf' return filename.encode('utf-8') -- 2.39.5