Warehouse: basic add / modify wizard

1 files changed, 18 insertions, 0 deletions

diff --git a/archaeological_warehouse/views.py b/archaeological_warehouse/views.py
index d8027a9a6..385a373e7 100644
--- a/archaeological_warehouse/views.py
+++ b/archaeological_warehouse/views.py
@@ -31,6 +31,8 @@ from forms import *
 
 get_container = get_item(models.Container, 'get_container', 'container')
 
+get_warehouse = get_item(models.Warehouse, 'get_warehouse', 'warehouse')
+
 new_warehouse = new_item(models.Warehouse, WarehouseForm)
 new_container = new_item(models.Container, ContainerForm)
 
@@ -87,6 +89,22 @@ warehouse_packaging_wizard = PackagingWizard.as_view([
     label=_(u"Packaging"),
     url_name='warehouse_packaging',)
 
+warehouse_creation_wizard = WarehouseWizard.as_view([
+    ("warehouse-warehouse_creation", WarehouseForm),
+    ('final-warehouse_creation', FinalForm)],
+    label=_(u"Warehouse creation"),
+    url_name='warehouse_creation',
+)
+
+warehouse_modification_wizard = WarehouseModificationWizard.as_view([
+    ('selec-warehouse_modification', WarehouseFormSelection),
+    ("warehouse-warehouse_modification", WarehouseForm),
+    ('final-warehouse_modification', FinalForm)],
+    label=_(u"Warehouse modification"),
+    url_name='warehouse_modification',
+)
+
+
 """
 warehouse_packaging_wizard = ItemSourceWizard.as_view([
          ('selec-warehouse_packaging', ItemsSelection),