From 0130ceb6b42e270832ee765e72b822c779eaf039 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Tue, 13 Dec 2016 19:03:20 +0100
Subject: Warehouse: basic add / modify wizard

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

(limited to 'archaeological_warehouse/wizards.py')

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