From da4af2ab5d105f6d2ce442b517e532b7570616e3 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 7 Apr 2017 10:47:39 +0200
Subject: Treatment - Treatment files dashboard: first overview (refs #3381)

---
 ishtar_common/models.py                                | 10 ++++------
 .../templates/ishtar/dashboards/dashboard_main.html    | 11 ++++++++++-
 ishtar_common/views.py                                 | 18 +++++++++++++++---
 3 files changed, 29 insertions(+), 10 deletions(-)

(limited to 'ishtar_common')

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index d8ba637db..988254359 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -2323,14 +2323,12 @@ class Import(models.Model):
     conservative_import = models.BooleanField(
         _(u"Conservative import"), default=False,
         help_text='If set to true, do not overload existing values')
-    creation_date = models.DateTimeField(_(u"Creation date"),
-                                         auto_now_add=True, blank=True,
-                                         null=True)
+    creation_date = models.DateTimeField(
+        _(u"Creation date"), auto_now_add=True, blank=True, null=True)
     end_date = models.DateTimeField(_(u"End date"), blank=True,
                                     null=True, editable=False)
-    seconds_remaining = models.IntegerField(_(u"Remaining seconds"),
-                                            blank=True, null=True,
-                                            editable=False)
+    seconds_remaining = models.IntegerField(
+        _(u"Remaining seconds"), blank=True, null=True, editable=False)
 
     class Meta:
         verbose_name = _(u"Import")
diff --git a/ishtar_common/templates/ishtar/dashboards/dashboard_main.html b/ishtar_common/templates/ishtar/dashboards/dashboard_main.html
index 93e11d604..6a5a67a63 100644
--- a/ishtar_common/templates/ishtar/dashboards/dashboard_main.html
+++ b/ishtar_common/templates/ishtar/dashboards/dashboard_main.html
@@ -15,7 +15,16 @@
 {% endblock %}
 {% block content %}
  <script>
-$(function() { $( "#dash-tabs" ).tabs(); });
+$(function() {
+ $( "#dash-tabs" ).tabs({
+   beforeActivate: function( event, ui ) {
+    $("#progress").show();
+   },
+   load: function( event, ui ) {
+    $("#progress").hide();
+   }
+ });
+});
 </script>
 <div id='dashboard'>
  <div id="dash-tabs">
diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 4732313ea..e8a2c9e12 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -1451,6 +1451,9 @@ def dashboard_main(request, dct, obj_id=None, *args, **kwargs):
         app_list.append((_(u"Context records"), 'contextrecords'))
     if profile.find:
         app_list.append((_(u"Finds"), 'finds'))
+    if profile.warehouse:
+        app_list.append((_(u"Treatment requests"), 'treatmentfiles'))
+        app_list.append((_(u"Treatments"), 'treatments'))
     dct = {'app_list': app_list}
     return render_to_response('ishtar/dashboards/dashboard_main.html', dct,
                               context_instance=RequestContext(request))
@@ -1497,18 +1500,27 @@ def dashboard_main_detail(request, item_name):
     if item_name == 'files' and profile.files:
         lbl, dashboard = (_(u"Archaeological files"),
                           models.Dashboard(File, **dashboard_kwargs))
-    if item_name == 'operations':
+    elif item_name == 'operations':
         from archaeological_operations.models import Operation
         lbl, dashboard = (_(u"Operations"),
                           models.Dashboard(Operation, **dashboard_kwargs))
-    if item_name == 'contextrecords' and profile.context_record:
+    elif item_name == 'contextrecords' and profile.context_record:
         lbl, dashboard = (
             _(u"Context records"),
             models.Dashboard(ContextRecord, slice=slicing, fltr=fltr))
-    if item_name == 'finds' and profile.find:
+    elif item_name == 'finds' and profile.find:
         lbl, dashboard = (_(u"Finds"), models.Dashboard(Find,
                                                         slice=slicing,
                                                         fltr=fltr))
+    elif item_name == 'treatmentfiles' and profile.warehouse:
+        lbl, dashboard = (
+            _(u"Treatment requests"),
+            models.Dashboard(TreatmentFile, slice=slicing, fltr=fltr))
+    elif item_name == 'treatments' and profile.warehouse:
+        lbl, dashboard = (
+            _(u"Treatments"),
+            models.Dashboard(Treatment, slice=slicing, fltr=fltr,
+                             date_source='start'))
     if not lbl:
         raise Http404
     dct = {'lbl': lbl, 'dashboard': dashboard,
-- 
cgit v1.2.3