fix(context-diagram): Make port processing more secure (#110)

DSD-DBS · Jun 19, 2024 · de72095 · de72095
1 parent a9b331d
commit de72095
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/capellambse_context_diagrams/collectors/default.py b/capellambse_context_diagrams/collectors/default.py
@@ -176,13 +176,10 @@ def _process_ports(self, stack_heights: dict[str, float | int]) -> None:
                 while (
                     current
                     and current.uuid not in self.diagram_target_owners
-                    and hasattr(current, "owner")
+                    and getattr(current, "owner", None) is not None
                     and not isinstance(current.owner, generic.PackageTypes)
                 ):
-                    current = self._make_owner_box(
-                        self.diagram,
-                        current,
-                    )
+                    current = self._make_owner_box(self.diagram, current)
                 self.common_owners.add(current.uuid)
 
             stack_heights[side] += makers.NEIGHBOR_VMARGIN + height