⚡ hierarchical type: improve full label

1 files changed, 23 insertions, 8 deletions

diff --git a/ishtar_common/models_common.py b/ishtar_common/models_common.py
index 0161aadad..4b0f1b160 100644
--- a/ishtar_common/models_common.py
+++ b/ishtar_common/models_common.py
@@ -803,14 +803,29 @@ class HierarchicalType(GeneralType):
     has_full_label = True
 
     def full_label(self):
-        lbls = [self.label]
-        item = self
-        parents = [self.pk]  # prevent loop
-        while item.parent and item.parent_id not in parents:
-            parents.append(item.parent_id)
-            item = item.parent
-            lbls.append(item.label)
-        return " > ".join(reversed(lbls))
+        return " > ".join(reversed(self.get_label_hierarchy()))
+
+    def _get_label_hierarchy(self, labels, pk, parents):
+        if not pk or pk in parents:
+            return labels
+        parents.add(pk)
+        q = self.__class__.objects.filter(pk=pk)
+        label, parent_id = q.values_list("label", "parent_id").all()[0]
+        labels.append(label)
+        self._get_label_hierarchy(labels, parent_id, parents)
+        return labels
+
+    def get_label_hierarchy(self, lower=False):
+        """
+        Get every upward labels of the hierarchy for full label or
+        cached hierarchical search.
+        Returned in reversed order.
+        """
+        parents = set((self.pk,))  # prevent loop
+        result = self._get_label_hierarchy([self.label], self.parent_id, parents)
+        if lower:
+            result = list(map(str.lower, result))
+        return result
 
     @property
     def first_parent(self):