Fix reset wizards

3 files changed, 59 insertions, 0 deletions

diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 7fa72a6b2..898610007 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -169,3 +169,14 @@ record_source_deletion_wizard = RecordSourceDeletionWizard.as_view([
     ('final-record_source_deletion', RecordDeletionForm)],
     label=_(u"Context record: source deletion"),
     url_name='record_source_deletion',)
+
+
+def reset_wizards(request):
+    for wizard_class, url_name in (
+            (RecordWizard, 'record_creation'),
+            (RecordModifWizard, 'record_modification'),
+            (RecordDeletionWizard, 'record_deletion'),
+            (RecordSourceWizard, 'record_source_creation'),
+            (RecordSourceWizard, 'record_source_modification'),
+            (RecordSourceDeletionWizard, 'record_source_deletion')):
+        wizard_class.session_reset(request, url_name)
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 75dda5d36..a2b497c9a 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -685,6 +685,7 @@ def treatmentfile_source_modify(request, pk):
         'treatmentfile_source_modification',
         kwargs={'step': 'source-treatmentfile_source_modification'}))
 
+
 treatmentfile_source_deletion_wizard = \
     TreatmentFileSourceDeletionWizard.as_view([
         ('selec-treatmentfile_source_deletion',
@@ -692,3 +693,37 @@ treatmentfile_source_deletion_wizard = \
         ('final-treatmentfile_source_deletion', SourceDeletionForm)],
         label=_(u"Treatment request: source deletion"),
         url_name='treatmentfile_source_deletion',)
+
+
+def reset_wizards(request):
+    for wizard_class, url_name in (
+            (FindWizard, 'find_creation'),
+            (FindModificationWizard, 'find_modification'),
+            (FindDeletionWizard, 'find_deletion'),
+            (FindSourceWizard, 'find_source_creation'),
+            (FindSourceWizard, 'find_source_modification'),
+            (FindSourceDeletionWizard, 'find_source_deletion'),
+            (TreatmentWizard, 'treatement_creation'),
+            (TreatmentModificationWizard, 'treatment_modification'),
+            (TreatmentDeletionWizard, 'treatment_deletion'),
+            (TreatmentAdministrativeActWizard, 'treatment_admacttreatment'),
+            (TreatmentEditAdministrativeActWizard,
+             'treatment_admacttreatment_modification'),
+            (TreatmentDeletionWizard, 'treatment_admacttreatment_deletion'),
+            (TreatmentFileWizard, 'treatmentfile_creation'),
+            (TreatmentFileModificationWizard, 'treatmentfile_modification'),
+            (TreatmentFileDeletionWizard, 'treatmentfile_deletion'),
+            (TreatmentFileAdministrativeActWizard,
+             'treatmentfle_admacttreatmentfle'),
+            (TreatmentFileEditAdministrativeActWizard,
+             'treatmentfle_admacttreatmentfle_modification'),
+            (AdministrativeActDeletionWizard,
+             'treatmentfle_admacttreatmentfle_deletion'),
+            (TreatmentSourceWizard, 'treatment_source_creation'),
+            (TreatmentSourceWizard, 'treatment_source_modification'),
+            (TreatmentSourceDeletionWizard, 'treatment_source_deletion'),
+            (TreatmentFileSourceWizard, 'treatmentfile_source_creation'),
+            (TreatmentFileSourceWizard, 'treatmentfile_source_modification'),
+            (TreatmentFileSourceDeletionWizard, 'treatmentfile_source_deletion')
+    ):
+        wizard_class.session_reset(request, url_name)
diff --git a/archaeological_warehouse/views.py b/archaeological_warehouse/views.py
index 60d31c09d..f33991d8b 100644
--- a/archaeological_warehouse/views.py
+++ b/archaeological_warehouse/views.py
@@ -183,3 +183,16 @@ warehouse_packaging_wizard = ItemSourceWizard.as_view([
          ('final-warehouse_packaging', FinalForm)],
           url_name='warehouse_packaging',)
 """
+
+
+def reset_wizards(request):
+    for wizard_class, url_name in (
+            (PackagingWizard, 'warehouse_packaging'),
+            (WarehouseWizard, 'warehouse_creation'),
+            (WarehouseModificationWizard, 'warehouse_modification'),
+            (WarehouseDeletionWizard, 'warehouse_deletion'),
+            (ContainerWizard, 'container_creation'),
+            (ContainerModificationWizard, 'container_modification'),
+            (ContainerDeletionWizard, 'container_deletion'),
+    ):
+        wizard_class.session_reset(request, url_name)