diff options
| -rw-r--r-- | archaeological_finds/views.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py index 5ec6fd89a..d04a15c67 100644 --- a/archaeological_finds/views.py +++ b/archaeological_finds/views.py @@ -656,13 +656,16 @@ treatment_creation_n1_wizard = wizards.TreatmentN1Wizard.as_view( def treatment_n1_create(request, pks): - if not wizard_is_available(treatment_creation_n1_wizard, models.Find, pks): - return HttpResponseRedirect("/") - wizards.TreatmentWizard.session_set_value( + for pk in pks.split("-"): + 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 ) return redirect( - "treatment_creation_n1", kwargs={"step": "file-treatment_creation_n1"} + reverse( + "treatment_creation_n1", kwargs={"step": "file-treatment_creation_n1"} + ) ) treatment_1n_wizard_steps = [ |