From e36ae6b8508a1e60f45a2edffb40e4b104e29de9 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Tue, 13 Dec 2016 11:12:16 +0100
Subject: Treatments: fix associated filename - add modify links in sheets

---
 archaeological_finds/views.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'archaeological_finds/views.py')

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)],
-- 
cgit v1.2.3