Find wizard: manage a simplified form for find with many base finds

1 files changed, 15 insertions, 1 deletions

diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py
index 3043d22aa..3c25258a6 100644
--- a/archaeological_finds/wizards.py
+++ b/archaeological_finds/wizards.py
@@ -55,11 +55,21 @@ class FindWizard(Wizard):
             if base_finds:
                 return base_finds[0].context_record
 
+    def get_current_basefinds(self):
+        find = self.get_current_object()
+        if not find:
+            return []
+        return find.base_finds.all()
+
     def get_form_kwargs(self, step=None):
         kwargs = super(FindWizard, self).get_form_kwargs(step)
-        if step not in ('find-find_creation', 'find-find_modification'):
+        if step not in (
+                'find-find_creation', 'find-find_modification',
+                'simplefind-find_modification',):
             return kwargs
         kwargs['context_record'] = self.get_current_contextrecord()
+        if step == 'simplefind-find_modification':
+            kwargs['base_finds'] = self.get_current_basefinds()
         return kwargs
 
     def get_context_data(self, form, **kwargs):
@@ -90,6 +100,10 @@ class FindModificationWizard(FindWizard):
         'selec-find_modification': ['pk'],
         'selecw-find_modification': ['pk'],
     }
+    wizard_templates = {
+        'simplefind-find_modification':
+            'ishtar/wizard/wizard_simplefind.html',
+    }
 
 
 class FindDeletionWizard(DeletionWizard):