From 474d8381ab18a29778b5786db7f4730daa32e570 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Wed, 10 Jul 2024 18:29:23 +0500 Subject: [PATCH] minor refactor --- src/blenderbim/blenderbim/bim/module/geometry/data.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/blenderbim/blenderbim/bim/module/geometry/data.py b/src/blenderbim/blenderbim/bim/module/geometry/data.py index 0a4bb17c40..b3a73f28d6 100644 --- a/src/blenderbim/blenderbim/bim/module/geometry/data.py +++ b/src/blenderbim/blenderbim/bim/module/geometry/data.py @@ -152,12 +152,12 @@ class RepresentationItemsData: cls.is_loaded = True @classmethod - def total_items(cls): - active_representation_id = None + def total_items(cls) -> int: result = 0 - if bpy.context.active_object.data and hasattr(bpy.context.active_object.data, "BIMMeshProperties"): - active_representation_id = bpy.context.active_object.data.BIMMeshProperties.ifc_definition_id - element = tool.Ifc.get().by_id(active_representation_id) + obj = bpy.context.active_object + assert obj + element = tool.Geometry.get_active_representation(obj) + if element: if not element.is_a("IfcShapeRepresentation"): return 0 queue = list(element.Items)