✨ improvements for templates

- add fields for document
- defaut inititialization (prevent crash when indexing a list)
- add a hierarchy variable for document source types

1 files changed, 16 insertions, 0 deletions

diff --git a/archaeological_warehouse/models.py b/archaeological_warehouse/models.py
index 35ef88147..093d56656 100644
--- a/archaeological_warehouse/models.py
+++ b/archaeological_warehouse/models.py
@@ -1322,6 +1322,22 @@ class Container(
     def natural_key(self):
         return (self.uuid,)
 
+    @property
+    def location_lbl(self):
+        return str(self.location or "")
+
+    @property
+    def responsability_lbl(self):
+        return str(self.responsibility or "")
+
+    def get_values(self, prefix="", no_values=False, filtr=None, **kwargs):
+        if "simple" in kwargs:
+            keys = ["reference", "container_type", "code", "cached_location",
+                    "cached_location", "index", "custom_index", "complete_identifier",
+                    "responsability_lbl", "location_lbl"]
+            return self._get_values(keys, prefix=prefix, filtr=filtr, **kwargs)
+        return super().get_values(prefix, no_values, filtr, **kwargs)
+
     @classmethod
     @pre_importer_action
     def import_get_location(cls, context, value):