From dafc097282ea0138f8bd7c0e8a705bd18c90b5df Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 24 Mar 2017 17:48:18 +0100
Subject: Warehouse: divisions forms is readonly if containers are localised in
 this warehouse (refs #3563)

---
 archaeological_warehouse/wizards.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'archaeological_warehouse/wizards.py')

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):
-- 
cgit v1.2.3