diff options
Diffstat (limited to 'archaeological_files_pdl/wizards.py')
| -rw-r--r-- | archaeological_files_pdl/wizards.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/archaeological_files_pdl/wizards.py b/archaeological_files_pdl/wizards.py index 666de8155..eb3611b26 100644 --- a/archaeological_files_pdl/wizards.py +++ b/archaeological_files_pdl/wizards.py @@ -18,6 +18,7 @@ # See the file COPYING for details. from archaeological_files.wizards import FileWizard as BaseFileWizard +from archaeological_files import models class FileWizard(BaseFileWizard): @@ -49,6 +50,21 @@ class FileWizard(BaseFileWizard): returned['year'] = self.get_current_year() return returned + def get_context_data(self, form, **kwargs): + context = super(FileWizard, self).get_context_data(form) + formkey = "planningservice-" + self.url_name + if self.steps.current == formkey: + try: + idx = int( + self.session_get_value( + 'preventivetype-' + self.url_name, 'permit_type')) + permit_type = models.PermitType.objects.get(pk=idx) + context['permit_type'] = unicode(permit_type) + context['permit_type_code'] = unicode(permit_type.txt_idx) + except (ValueError, models.PermitType.DoesNotExist): + pass + return context + class FileModificationWizard(FileWizard): modification = True |