Display every steps when modifying (refs #438)

1 files changed, 32 insertions, 6 deletions

diff --git a/ishtar/ishtar_base/forms.py b/ishtar/ishtar_base/forms.py
index e4f52b1cd..287a181f4 100644
--- a/ishtar/ishtar_base/forms.py
+++ b/ishtar/ishtar_base/forms.py
@@ -105,6 +105,7 @@ class SearchWizard(NamedUrlSessionFormWizard):
 
 class Wizard(NamedUrlSessionFormWizard):
     model = None
+    modification = None # True when the wizard modify an item
 
     def get_wizard_name(self):
         """
@@ -122,7 +123,7 @@ class Wizard(NamedUrlSessionFormWizard):
 
     def get_template_context(self, request, storage, form=None):
         """
-        Add previous and current steps to manage the wizard path
+        Add previous, next and current steps to manage the wizard path
         """
         context = super(Wizard, self).get_template_context(request, storage,
                                                            form)
@@ -132,13 +133,38 @@ class Wizard(NamedUrlSessionFormWizard):
         context.update({'current_step':self.form_list[current_step]})
         if step == current_step:
             return context
-        previous_steps = []
+        previous_steps, next_steps, previous_step_counter = [], [], 0
         while step:
             if step == current_step:
                 break
             previous_steps.append(self.form_list[step])
             step = self.get_next_step(request, storage, step)
-        context.update({'previous_steps':previous_steps})
+            previous_step_counter += 1
+        context.update({'previous_steps':previous_steps,
+                        'previous_step_counter':previous_step_counter})
+        # if modification: show the next steps
+        next_step = step
+        if self.modification:
+            while next_step:
+                # check if the form is initialized otherwise initialize it
+                if not storage.get_step_data(next_step):
+                    values = self.get_form_initial(request, storage, next_step)
+                    prefixed_values = {}
+                    if not isinstance(values, list):
+                        for key in values:
+                            form_key = next_step + '-' + key
+                            prefixed_values[form_key] = values[key]
+                    else:
+                        for formset_idx, v in enumerate(values):
+                            prefix = u"-%d-" % formset_idx
+                            for key in v:
+                                form_key = next_step + prefix + key
+                                prefixed_values[form_key] = v[key]
+                    storage.set_step_data(next_step, prefixed_values)
+                if step != next_step: # if not current step
+                    next_steps.append(self.form_list[next_step])
+                next_step = self.get_next_step(request, storage, next_step)
+        context.update({'next_steps':next_steps})
         # not last step: validation
         if step != self.get_last_step(request, storage):
             return context
@@ -582,12 +608,12 @@ class Wizard(NamedUrlSessionFormWizard):
                 return initial
         return super(Wizard, self).get_form_initial(request, storage, step)
 
-    def get_instanced_init(self, obj, request, storage, step):
+    def get_instanced_init(self, obj, request, storage, step=None):
         """
         Get initial data from an init
         """
-        current_step = storage.get_current_step() or self.get_first_step(
-                                                            request, storage)
+        current_step = step or  storage.get_current_step() \
+                       or self.get_first_step(request, storage)
         c_form = self.form_list[current_step]
         # make the current object the default item for the session
         obj_name = obj.__class__.__name__.lower()