Warehouse: divisions forms is readonly if containers are localised in this warehouse (refs #3563)

1 files changed, 14 insertions, 0 deletions

diff --git a/archaeological_warehouse/wizards.py b/archaeological_warehouse/wizards.py
index 9ecc16b3f..571e56b1b 100644
--- a/archaeological_warehouse/wizards.py
+++ b/archaeological_warehouse/wizards.py
@@ -64,6 +64,20 @@ class WarehouseModificationWizard(Wizard):
     model = models.Warehouse
     modification = True
     wizard_done_window = reverse_lazy('show-warehouse')
+    wizard_templates = {
+        'divisions-warehouse_modification':
+            'ishtar/wizard/wizard_warehouse_divisions.html',
+    }
+
+    def get_form_kwargs(self, step=None):
+        kwargs = super(WarehouseModificationWizard, self).get_form_kwargs(step)
+        if step == "divisions-warehouse_modification":
+            current_warehouse = self.get_current_object()
+            q = models.ContainerLocalisation.objects.filter(
+                    division__warehouse=current_warehouse)
+            if q.count():
+                kwargs['readonly'] = True
+        return kwargs
 
 
 class WarehouseDeletionWizard(DeletionWizard):