Change all upload directories

1 files changed, 12 insertions, 1 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index 048d3064d..866e218aa 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -92,7 +92,6 @@ class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
         "treatment_types__label": _(u"Type"),
         "treatment_state__label": _(u"State"),
     }
-    IMAGE_PREFIX = 'treatment'
     # extra keys than can be passed to save method
     EXTRA_SAVED_KEYS = ('items', 'user')
     SLUG = 'treatment'
@@ -202,6 +201,10 @@ class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
                  getattr(self, k)]
         return u'{} | {}'.format(u"-".join(items), self.treatment_types_lbl())
 
+    def _get_base_image_path(self,):
+        return u"treatment/{}/{}".format(
+            self.year, self.index)
+
     def treatment_types_lbl(self):
         """
         Treatment types label
@@ -631,6 +634,10 @@ class TreatmentSource(Source):
     def owner(self):
         return self.treatment
 
+    def _get_base_image_path(self):
+        return u"treatment/{}/{}/source".format(
+            self.treatment.year, self.treatment.index)
+
 
 class TreatmentFileSource(Source):
     treatment_file = models.ForeignKey(
@@ -660,3 +667,7 @@ class TreatmentFileSource(Source):
     @property
     def owner(self):
         return self.treatment_file
+
+    def _get_base_image_path(self):
+        return u"treatmentfile/{}/{}/source".format(
+            self.treatment_file.year, self.treatment_file.index)