From ee8a3927b431371667995d7cd1531c8c61f91000 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Mon, 24 Feb 2025 14:19:52 +0500 Subject: [PATCH] Fix displaying some attr types names after e8cb9e7 #6216 --- src/bonsai/bonsai/bim/prop.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/bonsai/bonsai/bim/prop.py b/src/bonsai/bonsai/bim/prop.py index e335af9a0e..9a69807c23 100644 --- a/src/bonsai/bonsai/bim/prop.py +++ b/src/bonsai/bonsai/bim/prop.py @@ -273,12 +273,16 @@ def set_length_value(self: "Attribute", value: float) -> None: def get_display_name(self: "Attribute") -> str: + DISPLAY_UNIT_TYPES = ("AREA", "VOLUME", "FORCE") name = self.name - if not self.special_type or self.special_type == "LENGTH": + if not self.special_type or self.special_type not in DISPLAY_UNIT_TYPES: return name unit_type = f"{self.special_type}UNIT" project_unit = ifcopenshell.util.unit.get_project_unit(tool.Ifc.get(), unit_type) + if not project_unit: + return name + unit_symbol = ifcopenshell.util.unit.get_unit_symbol(project_unit) return f"{name}, {unit_symbol}"