Warehouse: basic add / modify wizard

1 files changed, 12 insertions, 0 deletions

diff --git a/archaeological_warehouse/wizards.py b/archaeological_warehouse/wizards.py
index 62b441e00..f5aee5c1e 100644
--- a/archaeological_warehouse/wizards.py
+++ b/archaeological_warehouse/wizards.py
@@ -20,9 +20,12 @@
 from django.shortcuts import render_to_response
 from django.template import RequestContext
 
+from ishtar_common.wizards import Wizard
 from archaeological_finds.wizards import TreatmentWizard
 from archaeological_finds.models import Treatment
 
+import models
+
 
 class PackagingWizard(TreatmentWizard):
     basket_step = 'base-packaging'
@@ -40,3 +43,12 @@ class PackagingWizard(TreatmentWizard):
         res = render_to_response('ishtar/wizard/wizard_done.html', {},
                                  context_instance=RequestContext(self.request))
         return return_object and (obj, res) or res
+
+
+class WarehouseWizard(Wizard):
+    model = models.Warehouse
+
+
+class WarehouseModificationWizard(Wizard):
+    model = models.Warehouse
+    modification = True