Treament file dashboard: add filter form (refs #3381)

1 files changed, 7 insertions, 5 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 4aa452641..94e4c1582 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -63,7 +63,8 @@ from archaeological_finds.models import Find, Treatment, TreatmentFile, \
 
 from archaeological_operations.forms import DashboardForm as DashboardFormOpe
 from archaeological_files.forms import DashboardForm as DashboardFormFile
-from archaeological_finds.forms import DashboardTreatmentForm
+from archaeological_finds.forms import DashboardTreatmentForm, \
+    DashboardTreatmentFileForm
 
 from ishtar_common.forms import FinalForm, FinalDeleteForm
 from ishtar_common.widgets import JQueryAutoComplete
@@ -1461,7 +1462,8 @@ def dashboard_main(request, dct, obj_id=None, *args, **kwargs):
 
 DASHBOARD_FORMS = {
     'files': DashboardFormFile, 'operations': DashboardFormOpe,
-    'treatments': DashboardTreatmentForm
+    'treatments': DashboardTreatmentForm,
+    'treatmentfiles': DashboardTreatmentFileForm
 }
 
 
@@ -1478,9 +1480,9 @@ def dashboard_main_detail(request, item_name):
     slicing, date_source, fltr, show_detail = 'year', None, {}, False
     profile = models.get_current_profile()
     has_form = (item_name == 'files' and profile.files) \
-            or item_name == 'operations' \
-            or (item_name in ('treatmentfiles', 'treatments')
-                and profile.warehouse)
+        or item_name == 'operations' \
+        or (item_name in ('treatmentfiles', 'treatments')
+            and profile.warehouse)
     if has_form:
         slicing = 'month'
     if item_name in DASHBOARD_FORMS: