Archaeological files: reference_number -> string - better layout for planning service form

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