From be2f9250a3684d8a7b42cdedcb97b88689146583 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Wed, 22 Jun 2016 12:02:48 +0200
Subject: Add link to modify sources in the sheet

---
 archaeological_operations/models.py | 1 +
 archaeological_operations/urls.py   | 2 ++
 archaeological_operations/views.py  | 9 +++++++++
 3 files changed, 12 insertions(+)

(limited to 'archaeological_operations')

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 6ae05ca30..0ae042bf0 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -517,6 +517,7 @@ class OperationSource(Source):
     TABLE_COLS = ['operation.year', 'operation.operation_code'] + \
         Source.TABLE_COLS
     SHOW_URL = 'show-operationsource'
+    MODIFY_URL = 'operation_source_modify'
 
     @property
     def owner(self):
diff --git a/archaeological_operations/urls.py b/archaeological_operations/urls.py
index ef7d14dd6..b25390bca 100644
--- a/archaeological_operations/urls.py
+++ b/archaeological_operations/urls.py
@@ -56,6 +56,8 @@ urlpatterns = patterns(
         check_rights(['change_operation', 'change_own_operation'])(
             views.operation_source_modification_wizard),
         name='operation_source_modification'),
+    url(r'operation_source_modify/(?P<pk>.+)/$',
+        views.operation_source_modify, name='operation_source_modify'),
     url(r'operation_source_deletion/(?P<step>.+)?$',
         check_rights(['change_operation', 'change_own_operation'])(
             views.operation_source_deletion_wizard),
diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py
index 0b9b9c453..14160266b 100644
--- a/archaeological_operations/views.py
+++ b/archaeological_operations/views.py
@@ -361,6 +361,15 @@ operation_source_modification_wizard = OperationSourceWizard.as_view([
     label=_(u"Operation: source modification"),
     url_name='operation_source_modification',)
 
+
+def operation_source_modify(request, pk):
+    operation_source_modification_wizard(request)
+    OperationSourceWizard.session_set_value(
+        request, 'selec-operation_source_modification', 'pk', pk, reset=True)
+    return redirect(reverse(
+        'operation_source_modification',
+        kwargs={'step': 'source-operation_source_modification'}))
+
 operation_source_deletion_wizard = OperationSourceDeletionWizard.as_view([
     ('selec-operation_source_deletion', OperationSourceFormSelection),
     ('final-operation_source_deletion', SourceDeletionForm)],
-- 
cgit v1.2.3