QA sheet: administrative act action

1 files changed, 21 insertions, 3 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index 15ef64040..080d2cc50 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -267,6 +267,17 @@ class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
     upstream_lbl.short_description = _(u"Upstream finds")
     upstream_lbl.admin_order_field = 'upstream__cached_label'
 
+    def get_extra_actions(self, request):
+        # url, base_text, icon, extra_text, extra css class, is a quick action
+        actions = []
+        if self.can_do(request, 'add_administrativeact'):
+            actions += [
+                (reverse('treatment-add-adminact', args=[self.pk]),
+                 _(u"Add associated administrative act"), "fa fa-plus",
+                 _(u"admin. act"), "", False),
+            ]
+        return actions
+
     def get_values(self, prefix=''):
         values = super(Treatment, self).get_values(prefix=prefix)
         values[prefix + "upstream_finds"] = u" ; ".join(
@@ -843,13 +854,20 @@ class TreatmentFile(DashboardFormItem, ClosedItem, BaseHistorizedItem,
                                       'name') if getattr(self, attr)])
 
     def get_extra_actions(self, request):
+        # url, base_text, icon, extra_text, extra css class, is a quick action
+        actions = []
+        if self.can_do(request, 'add_administrativeact'):
+            actions += [
+                (reverse('treatmentfile-add-adminact', args=[self.pk]),
+                 _(u"Add associated administrative act"), "fa fa-plus",
+                 _(u"admin. act"), "", False),
+            ]
         if not self.associated_basket:
-            return []
+            return actions
         if self.type.treatment_type and self.treatments.filter(
                 treatment_types__pk=self.type.treatment_type.pk).count():
             # a treatment of this type already exists
-            return []
-        actions = []
+            return actions
         can_edit_find = self.can_do(request, 'change_find')
         if can_edit_find:
             actions += [