Document generation: WIP on basket

1 files changed, 9 insertions, 0 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index 080d2cc50..3d3827596 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -853,6 +853,15 @@ class TreatmentFile(DashboardFormItem, ClosedItem, BaseHistorizedItem,
                          for attr in ('year', 'index', 'internal_reference',
                                       'name') if getattr(self, attr)])
 
+    def get_values(self, prefix=''):
+        values = super(TreatmentFile, self).get_values(prefix=prefix)
+        if not self.associated_basket:
+            return
+        values[prefix + "basket"] = [
+            find.get_values() for find in self.associated_basket.items.all()
+        ]
+        return values
+
     def get_extra_actions(self, request):
         # url, base_text, icon, extra_text, extra css class, is a quick action
         actions = []