Add link to modify sources in the sheet

3 files changed, 13 insertions, 1 deletions

diff --git a/archaeological_context_records/models.py b/archaeological_context_records/models.py
index ddd9c8609..e449e893d 100644
--- a/archaeological_context_records/models.py
+++ b/archaeological_context_records/models.py
@@ -313,6 +313,7 @@ post_delete.connect(post_delete_record_relation, sender=RecordRelations)
 
 class ContextRecordSource(Source):
     SHOW_URL = 'show-contextrecordsource'
+    MODIFY_URL = 'record_source_modify'
 
     class Meta:
         verbose_name = _(u"Context record documentation")
diff --git a/archaeological_context_records/urls.py b/archaeological_context_records/urls.py
index 52af8bda9..84560da67 100644
--- a/archaeological_context_records/urls.py
+++ b/archaeological_context_records/urls.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2010-2013 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2010-2016 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as
@@ -51,6 +51,8 @@ urlpatterns = patterns(
         check_rights(['change_contextrecord', 'change_own_contextrecord'])(
             views.record_source_modification_wizard),
         name='record_source_modification'),
+    url(r'record_source_modify/(?P<pk>.+)/$',
+        views.record_source_modify, name='record_source_modify'),
     url(r'record_source_deletion/(?P<step>.+)?$',
         check_rights(['change_contextrecord', 'change_own_contextrecord'])(
             views.record_source_deletion_wizard),
diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 839dbed9e..c76413977 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -161,6 +161,15 @@ record_source_modification_wizard = RecordSourceWizard.as_view([
     label=_(u"Context record: source modification"),
     url_name='record_source_modification',)
 
+
+def record_source_modify(request, pk):
+    record_source_modification_wizard(request)
+    RecordSourceWizard.session_set_value(
+        request, 'selec-record_source_modification', 'pk', pk, reset=True)
+    return redirect(reverse(
+        'record_source_modification',
+        kwargs={'step': 'source-record_source_modification'}))
+
 record_source_deletion_wizard = RecordSourceDeletionWizard.as_view([
     ('selec-record_source_deletion', RecordSourceFormSelection),
     ('final-record_source_deletion', RecordDeletionForm)],