Archaeological files: on creation, final step display potential similar files (check based on parcels and towns)

1 files changed, 20 insertions, 0 deletions

diff --git a/archaeological_files_pdl/wizards.py b/archaeological_files_pdl/wizards.py
index 1d8351176..61294e538 100644
--- a/archaeological_files_pdl/wizards.py
+++ b/archaeological_files_pdl/wizards.py
@@ -37,6 +37,7 @@ class FileWizard(BaseFileWizard):
         'preventiveplanning-%(url_name)s':
             'ishtar/wizard/wizard_preventiveplanning.html',
     }
+    wizard_confirm = 'ishtar/wizard/file_confirm_wizard.html'
 
     def get_current_year(self):
         general_form_key = 'general-' + self.url_name
@@ -68,6 +69,7 @@ class FileWizard(BaseFileWizard):
         context = super(FileWizard, self).get_context_data(form)
         formplanning = "planningservice-" + self.url_name
         forminstruction = "instruction-" + self.url_name
+        formfinal = "final-" + self.url_name
         if self.steps.current == formplanning:
             try:
                 idx = int(
@@ -83,6 +85,24 @@ class FileWizard(BaseFileWizard):
             if saisine_type:
                 context['saisine_type'] = unicode(saisine_type)
                 context['saisine_type_delay'] = saisine_type.delay or 0
+        elif self.steps.current == formfinal and \
+                self.steps.current.endswith('creation'):  # creation only
+            parcels = []
+            parcel_step_key = self.parcel_step_key + self.url_name
+
+            parcel_numbers = self.session_get_value(
+                parcel_step_key, 'parcel_number', multi=True) or []
+            sections = self.session_get_value(
+                parcel_step_key, 'section', multi=True) or []
+            towns = self.session_get_value(
+                parcel_step_key, 'town', multi=True) or []
+            for idx, parcel_number in enumerate(parcel_numbers):
+                if not parcel_number:
+                    continue
+                parcels.append({'town': towns[idx],
+                        'section': sections[idx],
+                        'parcel_number': parcel_number})
+            context['similar_files'] = models.File.similar_files(parcels)
         return context