Fix reset wizards

1 files changed, 35 insertions, 0 deletions

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)