summaryrefslogtreecommitdiff
path: root/archaeological_finds/models_treatments.py
diff options
context:
space:
mode:
Diffstat (limited to 'archaeological_finds/models_treatments.py')
-rw-r--r--archaeological_finds/models_treatments.py20
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)