Treatments: fix pdf and odt exports

1 files changed, 8 insertions, 1 deletions

diff --git a/archaeological_finds/models.py b/archaeological_finds/models.py
index a563a8404..fc231a16a 100644
--- a/archaeological_finds/models.py
+++ b/archaeological_finds/models.py
@@ -24,6 +24,7 @@ from django.contrib.gis.db import models
 from django.core.urlresolvers import reverse
 from django.db.models import Max, Q
 from django.db.models.signals import m2m_changed, post_save, post_delete
+from django.template.defaultfilters import slugify
 from django.utils.translation import ugettext_lazy as _, ugettext
 
 from ishtar_common.utils import cached_label_changed
@@ -856,7 +857,7 @@ class Treatment(BaseHistorizedItem, ImageModel, OwnPerms):
         )
 
     def __unicode__(self):
-        lbl = unicode(self.treatment_type)
+        lbl = unicode(self.treatment_types_lbl())
         if self.person:
             lbl += u" %s %s" % (_(u"by"), unicode(self.person))
         return lbl
@@ -913,6 +914,12 @@ class Treatment(BaseHistorizedItem, ImageModel, OwnPerms):
                     find.skip_history_when_saving = True
                 find.save()
 
+    @property
+    def associated_filename(self):
+        return "-".join([
+            str(slugify(getattr(self, attr)))
+            for attr in ('year', 'index', 'label')])
+
 
 class AbsFindTreatments(models.Model):
     find = models.ForeignKey(Find, verbose_name=_(u"Find"),