Reorganize treatment actions - add a direct to treatements

1 files changed, 9 insertions, 0 deletions

diff --git a/archaeological_warehouse/wizards.py b/archaeological_warehouse/wizards.py
index 41c5c5945..8ee2c2972 100644
--- a/archaeological_warehouse/wizards.py
+++ b/archaeological_warehouse/wizards.py
@@ -38,6 +38,15 @@ class ContainerSearch(SearchWizard):
 class PackagingWizard(TreatmentWizard):
     basket_step = 'base-packaging'
 
+    def get_form_initial(self, step, data=None):
+        initial = super(PackagingWizard, self).get_form_initial(step)
+        user = self.request.user
+        if step != 'base-packaging' or not getattr(user, 'ishtaruser', None) \
+                or not user.ishtaruser.person:
+            return initial
+        initial['person'] = user.ishtaruser.person.pk
+        return initial
+
     def save_model(self, dct, m2m, whole_associated_models, form_list,
                    return_object):
         dct = self.get_extra_model(dct, form_list)