From 63ec480b4e617157d7bbb22820272496f14bb771 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 14 Nov 2018 18:02:04 +0100 Subject: Reorganize treatment actions - add a direct to treatements --- archaeological_warehouse/wizards.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'archaeological_warehouse/wizards.py') 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) -- cgit v1.2.3