Admin: improve forms, lists for finds, base finds, treatements, treatment requests, etc.

1 files changed, 20 insertions, 0 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index d1b1afd4b..048d3064d 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -208,6 +208,26 @@ class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
         :return: string
         """
         return u" ; ".join([unicode(t) for t in self.treatment_types.all()])
+    treatment_types_lbl.short_description = _(u"Treatment types")
+    treatment_types_lbl.admin_order_field = 'treatment_types__label'
+
+    def downstream_lbl(self):
+        """
+        Downstream finds label
+        :return: string
+        """
+        return u" ; ".join([f.cached_label for f in self.downstream.all()])
+    downstream_lbl.short_description = _(u"Downstream finds")
+    downstream_lbl.admin_order_field = 'downstream__cached_label'
+
+    def upstream_lbl(self):
+        """
+        Upstream finds label
+        :return: string
+        """
+        return u" ; ".join([f.cached_label for f in self.upstream.all()])
+    upstream_lbl.short_description = _(u"Upstream finds")
+    upstream_lbl.admin_order_field = 'upstream__cached_label'
 
     def get_values(self, prefix=''):
         values = super(Treatment, self).get_values(prefix=prefix)