Operation, site: multi del - redirs

1 files changed, 6 insertions, 5 deletions

diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py
index d71bd0d02..8bc805b7a 100644
--- a/archaeological_operations/views.py
+++ b/archaeological_operations/views.py
@@ -40,7 +40,8 @@ from archaeological_operations.forms import ArchaeologicalSiteForm, \
     AdministrativeActOpeModifForm, FinalAdministrativeActDeleteForm, \
     AdministrativeActRegisterFormSelection, DocumentGenerationAdminActForm, \
     SiteForm, SiteTownFormset, SiteUnderwaterForm, check_underwater_module, \
-    CourtOrderedSeizureForm, SiteSelect, OperationSelect, QAOperationFormMulti
+    CourtOrderedSeizureForm, SiteSelect, OperationSelect, \
+    QAOperationFormMulti, OperationFormMultiSelection, SiteFormMultiSelection
 from archaeological_operations.wizards import has_associated_file, \
     is_preventive, is_judiciary, OperationWizard, OperationModificationWizard, \
     OperationClosingWizard, OperationDeletionWizard, SiteSearch, \
@@ -332,7 +333,7 @@ operation_closing_wizard = OperationClosingWizard.as_view(
 
 
 operation_deletion_steps = [
-    ('selec-operation_deletion', OperationFormSelection),
+    ('selec-operation_deletion', OperationFormMultiSelection),
     ('final-operation_deletion', OperationDeletionForm)
 ]
 
@@ -349,7 +350,7 @@ def operation_delete(request, pk):
 
     wizard_url = 'operation_deletion'
     OperationDeletionWizard.session_set_value(
-        request, 'selec-' + wizard_url, 'pk', pk, reset=True)
+        request, 'selec-' + wizard_url, 'pks', pk, reset=True)
     return redirect(reverse(wizard_url,
                             kwargs={'step': 'final-' + wizard_url}))
 
@@ -414,7 +415,7 @@ def site_modify(request, pk):
 
 
 site_deletion_steps = [
-    ('selec-site_deletion', SiteFormSelection),
+    ('selec-site_deletion', SiteFormMultiSelection),
     ('final-site_deletion', FinalDeleteForm)
 ]
 
@@ -430,7 +431,7 @@ def site_delete(request, pk):
         return HttpResponseRedirect("/")
     wizard_url = 'site_deletion'
     SiteDeletionWizard.session_set_value(
-        request, 'selec-' + wizard_url, 'pk', pk, reset=True)
+        request, 'selec-' + wizard_url, 'pks', pk, reset=True)
     return redirect(reverse(wizard_url,
                             kwargs={'step': 'final-' + wizard_url}))