diff --git a/src/bonsai/bonsai/bim/module/classification/data.py b/src/bonsai/bonsai/bim/module/classification/data.py index 424e3f790e..b822903431 100644 --- a/src/bonsai/bonsai/bim/module/classification/data.py +++ b/src/bonsai/bonsai/bim/module/classification/data.py @@ -27,7 +27,7 @@ from bonsai.bim.ifc import IfcStore def refresh(): ClassificationsData.is_loaded = False - ClassificationReferencesData.is_loaded = False + ObjectClassificationsData.is_loaded = False MaterialClassificationsData.is_loaded = False CostClassificationsData.is_loaded = False @@ -92,7 +92,7 @@ class ReferencesData: return [(str(e.id()), e.Name, "") for e in tool.Ifc.get().by_type("IfcClassification")] -class ClassificationReferencesData(ReferencesData): +class ObjectClassificationsData(ReferencesData): data: dict[str, Any] = {} is_loaded = False diff --git a/src/bonsai/bonsai/bim/module/classification/prop.py b/src/bonsai/bonsai/bim/module/classification/prop.py index c6b12d9893..d85e4e8c4e 100644 --- a/src/bonsai/bonsai/bim/module/classification/prop.py +++ b/src/bonsai/bonsai/bim/module/classification/prop.py @@ -19,7 +19,7 @@ import bpy import bonsai.tool as tool from bonsai.bim.prop import Attribute -from bonsai.bim.module.classification.data import ClassificationsData, ClassificationReferencesData +from bonsai.bim.module.classification.data import ClassificationsData, ObjectClassificationsData from bpy.types import PropertyGroup from bpy.props import ( PointerProperty, @@ -43,9 +43,9 @@ def get_available_classifications( def get_classifications(self, context): - if not ClassificationReferencesData.is_loaded: - ClassificationReferencesData.load() - return ClassificationReferencesData.data["classifications"] + if not ObjectClassificationsData.is_loaded: + ObjectClassificationsData.load() + return ObjectClassificationsData.data["classifications"] def get_classification_source( diff --git a/src/bonsai/bonsai/bim/module/classification/ui.py b/src/bonsai/bonsai/bim/module/classification/ui.py index 41f638c93b..18038b539e 100644 --- a/src/bonsai/bonsai/bim/module/classification/ui.py +++ b/src/bonsai/bonsai/bim/module/classification/ui.py @@ -24,7 +24,7 @@ import bonsai.bim.module.classification.prop as classification_prop from bpy.types import Panel, UIList from bonsai.bim.module.classification.data import ( ClassificationsData, - ClassificationReferencesData, + ObjectClassificationsData, MaterialClassificationsData, CostClassificationsData, ) @@ -115,7 +115,7 @@ class ReferenceUI: layout: bpy.types.UILayout data: type[ Union[ - ClassificationReferencesData, + ObjectClassificationsData, MaterialClassificationsData, CostClassificationsData, ] @@ -266,9 +266,9 @@ class BIM_PT_classification_references(Panel, ReferenceUI): return bool(tool.Ifc.get_entity(context.active_object)) def draw(self, context): - if not ClassificationReferencesData.is_loaded: - ClassificationReferencesData.load() - self.data = ClassificationReferencesData + if not ObjectClassificationsData.is_loaded: + ObjectClassificationsData.load() + self.data = ObjectClassificationsData self.obj = context.active_object.name self.obj_type = "Object" self.draw_ui(context)