Add a new action to create an associated operation from a file window

1 files changed, 10 insertions, 1 deletions

diff --git a/archaeological_files/views.py b/archaeological_files/views.py
index ba640a785..c414ba858 100644
--- a/archaeological_files/views.py
+++ b/archaeological_files/views.py
@@ -28,6 +28,7 @@ from django.utils.translation import ugettext_lazy as _
 from ishtar_common.views import get_item, show_item, revert_item
 
 from ishtar_common.models import Person
+from archaeological_operations.models import Operation
 import models
 
 from ishtar_common.wizards import SearchWizard
@@ -85,9 +86,17 @@ get_file = get_item(models.File, 'get_file', 'file',
         'general_contractor__attached_to':'general_contractor__attached_to__pk',
         'history_creator':'history_creator__ishtaruser__person__pk',
             },)
-show_file = show_item(models.File, 'file')
 revert_file = revert_item(models.File)
 
+def extra_file_dct(request, item):
+    dct = {}
+    if (request.user.has_perm('ishtar_common.add_operation', Operation)
+       or request.user.ishtaruser.has_right('add_operation')):
+        dct['can_add_operation'] = True
+    return dct
+
+show_file = show_item(models.File, 'file', extra_dct=extra_file_dct)
+
 get_administrativeactfile = get_item(AdministrativeAct,
         'get_administrativeactfile', 'administrativeactfile',
       associated_models = [(models.File, 'associated_file'),