Find - Quick action n1 treatment: add redirect in table quick action - fix resulting_pk initialization (refs #5231)

1 files changed, 13 insertions, 16 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index d04a15c67..4def2e14b 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -353,10 +353,7 @@ find_creation_wizard = wizards.FindWizard.as_view(
 
 def find_create(request, pk):
     context_record = wizard_is_available(
-        find_creation_wizard,
-        request,
-        ContextRecord,
-        pk
+        find_creation_wizard, request, ContextRecord, pk
     )
     if not context_record:
         return HttpResponseRedirect("/")
@@ -656,18 +653,24 @@ treatment_creation_n1_wizard = wizards.TreatmentN1Wizard.as_view(
 
 
 def treatment_n1_create(request, pks):
-    for pk in pks.split("-"):
-        if not wizard_is_available(treatment_creation_n1_wizard, request, models.Find, pk):
+    pks = pks.split("-")
+    for pk in pks:
+        if not wizard_is_available(
+            treatment_creation_n1_wizard, request, models.Find, pk
+        ):
             return HttpResponseRedirect("/")
     wizards.TreatmentN1Wizard.session_set_value(
-        request, "selecfind-treatment_creation_n1", "pk", pks, reset=True
+        request,
+        "selecfind-treatment_creation_n1",
+        "resulting_pk",
+        ",".join(pks),
+        reset=True,
     )
     return redirect(
-        reverse(
-            "treatment_creation_n1", kwargs={"step": "file-treatment_creation_n1"}
-        )
+        reverse("treatment_creation_n1", kwargs={"step": "file-treatment_creation_n1"})
     )
 
+
 treatment_1n_wizard_steps = [
     ("selecfind-treatment_creation_1n", forms.SingleUpstreamFindFormSelection),
     ("file-treatment_creation_1n", forms.TreatmentFormFileChoice),
@@ -1321,12 +1324,6 @@ class QAFindLockView(QABaseLockView):
     base_url = "find-qa-lock"
 
 
-class QAFindTreatmentN1View(QAItemForm):
-    model = models.Treatment
-    form_class = forms.QAFindTreatmentN1Form
-    base_url = "find-qa-treatmentN1"
-
-
 def get_geo_items(request, get_polygons, current_right=None):
     operation_pk = request.GET.get("operation_pk")
     context_record_pk = request.GET.get("context_record_pk")