Sheet files: add operation shortcut

1 files changed, 16 insertions, 0 deletions

diff --git a/archaeological_files/views.py b/archaeological_files/views.py
index a29ae8b15..f94de1cb1 100644
--- a/archaeological_files/views.py
+++ b/archaeological_files/views.py
@@ -37,6 +37,8 @@ from archaeological_files import wizards
 from archaeological_files.wizards import FileWizard, \
     FileModificationWizard, FileClosingWizard, FileDeletionWizard, \
     FileAdministrativeActWizard, FileEditAdministrativeActWizard
+from archaeological_operations.wizards import OperationWizard
+from archaeological_operations.views import operation_creation_wizard
 
 from ishtar_common.forms_common import TownFormset
 from archaeological_operations.forms import FinalAdministrativeActDeleteForm
@@ -259,6 +261,20 @@ def file_adminact_add(request, pk, current_right=None):
         kwargs={'step': 'administrativeact-file_administrativeactfile'}))
 
 
+def file_add_operation(request, pk, current_right=None):
+    try:
+        models.File.objects.get(pk=pk)
+    except models.File.DoesNotExist:
+        raise Http404()
+    operation_creation_wizard(request)
+    OperationWizard.session_set_value(
+        request, 'filechoice-operation_creation', 'associated_file', pk,
+        reset=True)
+    return redirect(reverse(
+        'operation_creation',
+        kwargs={'step': 'general-operation_creation'}))
+
+
 def reset_wizards(request):
     for wizard_class, url_name in (
             (FileWizard, 'file_creation'),