fix: Fix-tree-view

Merge pull request #74 from DSD-DBS/fix-tree-view
DSD-DBS · Feb 13, 2024 · 5cb386e · 5cb386e
2 parents 605a6ea + de878b1
commit 5cb386e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/capellambse_context_diagrams/collectors/tree_view.py b/capellambse_context_diagrams/collectors/tree_view.py
@@ -40,6 +40,10 @@ def __init__(
         self.legend_boxes: list[_elkjs.ELKInputChild] = []
         self.all_associations = all_associations
 
+    def __contains__(self, uuid: str) -> bool:
+        objects = self.data["children"] + self.data["edges"]  # type: ignore[operator]
+        return uuid in {obj["id"] for obj in objects}
+
     def process_class(self, cls, params):
         self._process_box(cls.source, cls.partition, params)
 
@@ -108,7 +112,9 @@ def _set_data_types_and_labels(
         box["width"], box["height"] = makers.calculate_height_and_width(
             list(box["labels"])
         )
-        self.legend_boxes.extend(legends)
+        for legend in legends:
+            if legend["id"] not in self:
+                self.legend_boxes.append(legend)
 
 
 def collector(

diff --git a/docs/realization_view.md b/docs/realization_view.md
@@ -3,7 +3,7 @@
  ~ SPDX-License-Identifier: Apache-2.0
  -->
 
-# 🔥Brand-new🔥 Tree View Diagram 🔥Brand-new🔥
+# Tree View Diagram
 
 With release
 [`v0.5.42`](https://github.com/DSD-DBS/py-capellambse/releases/tag/v0.5.42) of

diff --git a/mkdocs.yml b/mkdocs.yml
@@ -91,7 +91,7 @@ nav:
       - Styling: extras/styling.md
   - Tree View:
       - Overview: tree_view.md
-  - Realization View:
+  - 🔥Realization View🔥:
       - Overview: realization_view.md
   - Code Reference: reference/