Document templates: manage finds, base finds, baskets relations

1 files changed, 10 insertions, 0 deletions

diff --git a/archaeological_context_records/models.py b/archaeological_context_records/models.py
index 01595feed..f728aef57 100644
--- a/archaeological_context_records/models.py
+++ b/archaeological_context_records/models.py
@@ -520,6 +520,16 @@ class ContextRecord(BulkUpdatedItem, BaseHistorizedItem, QRCodeItem, GeoItem,
     def __unicode__(self):
         return self.short_label
 
+    def get_values(self, prefix='', no_base_finds=True):
+        values = super(ContextRecord, self).get_values(prefix=prefix)
+        if prefix and no_base_finds:
+            return values
+        values[prefix + 'base_finds'] = [
+            bf.get_values(prefix=prefix)
+            for bf in self.base_finds.distinct().all()
+        ]
+        return values
+
     def get_town_centroid(self):
         if self.town:
             return self.town.center, self._meta.verbose_name