Treatments: fix associated filename - add modify links in sheets

1 files changed, 19 insertions, 0 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 2b5aef79c..01e579566 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -384,6 +384,16 @@ treatment_modification_wizard = TreatmentModificationWizard.as_view(
     url_name='treatment_modification',
 )
 
+
+def treatment_modify(request, pk):
+    treatment_modification_wizard(request)
+    TreatmentModificationWizard.session_set_value(
+        request, 'selec-treatment_modification', 'pk', pk, reset=True)
+    return redirect(reverse(
+        'treatment_modification',
+        kwargs={'step': 'basetreatment-treatment_modification'}))
+
+
 treatment_deletion_wizard = TreatmentDeletionWizard.as_view([
     ('selec-treatment_deletion', TreatmentFormSelection),
     ('final-treatment_deletion', TreatmentDeletionForm)],
@@ -466,6 +476,15 @@ treatmentfile_modification_wizard = TreatmentFileModificationWizard.as_view(
     url_name='treatmentfile_modification',
 )
 
+
+def treatmentfile_modify(request, pk):
+    treatmentfile_modification_wizard(request)
+    TreatmentFileModificationWizard.session_set_value(
+        request, 'selec-treatmentfile_modification', 'pk', pk, reset=True)
+    return redirect(reverse(
+        'treatmentfile_modification',
+        kwargs={'step': 'treatmentfile-treatmentfile_modification'}))
+
 treatmentfile_deletion_wizard = TreatmentFileDeletionWizard.as_view([
     ('selec-treatmentfile_deletion', TreatmentFileFormSelection),
     ('final-treatmentfile_deletion', TreatmentFileDeletionForm)],