Add administrative index to finds

1 files changed, 24 insertions, 1 deletions

diff --git a/archaeological_finds/models.py b/archaeological_finds/models.py
index 99a7f14cb..36702b707 100644
--- a/archaeological_finds/models.py
+++ b/archaeological_finds/models.py
@@ -257,6 +257,7 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
     mark = models.TextField(_(u"Mark"), blank=True, null=True)
     comment = models.TextField(_(u"Comment"), blank=True, null=True)
     previous_id = models.TextField(_(u"Previous ID"), blank=True, null=True)
+    index = models.IntegerField(u"Index", default=0)
     history = HistoricalRecords()
 
     def __init__(self, *args, **kwargs):
@@ -301,7 +302,7 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
         q = self.base_finds
         if not q.count():
             return
-        return q.all()[0]
+        return q.order_by('-pk').all()[0]
 
     @property
     def reference(self):
@@ -310,6 +311,15 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
             return "00"
         return bf.short_id()
 
+    @property
+    def administrative_index(self):
+        bf = self.get_first_base_find()
+        if not bf:
+            return
+        return "{}-{}".format(
+            bf.context_record.operation.get_reference(),
+            self.index)
+
     def get_department(self):
         bf = self.get_first_base_find()
         if not bf:
@@ -391,6 +401,19 @@ class Find(BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
 
     def save(self, *args, **kwargs):
         super(Find, self).save(*args, **kwargs)
+        q = self.base_finds
+        if not self.index and q.count():
+            # TODO: which base_find to take?
+            operation = q.all()[0].context_record.operation
+            q = Find.objects\
+                .filter(base_finds__context_record__operation=operation)
+            if self.pk:
+                q = q.exclude(pk=self.pk)
+            if q.count():
+                self.index = q.aggregate(Max('index'))['index__max'] + 1
+            else:
+                self.index = 1
+            self.save()
         for base_find in self.base_finds.all():
             if not base_find.index:
                 idx = BaseFind.objects\