From 667223da8e7e81f8b6344d95c71e78e0f296bfdb Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Wed, 5 Feb 2025 18:40:02 +0500 Subject: [PATCH] typing --- src/bonsai/bonsai/bim/module/pset/prop.py | 1 + src/bonsai/bonsai/bim/module/root/operator.py | 4 ++-- src/bonsai/bonsai/bim/module/root/prop.py | 1 + src/bonsai/bonsai/tool/pset.py | 1 + 4 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/bonsai/bonsai/bim/module/pset/prop.py b/src/bonsai/bonsai/bim/module/pset/prop.py index 3f73fa4acd..1edf364167 100644 --- a/src/bonsai/bonsai/bim/module/pset/prop.py +++ b/src/bonsai/bonsai/bim/module/pset/prop.py @@ -241,6 +241,7 @@ class IfcProperty(PropertyGroup): if TYPE_CHECKING: metadata: Attribute value_type: IfcPropertyValueType + enumerated_value: IfcPropertyEnumeratedValue class PsetProperties(PropertyGroup): diff --git a/src/bonsai/bonsai/bim/module/root/operator.py b/src/bonsai/bonsai/bim/module/root/operator.py index 9727556115..9c82d71c97 100644 --- a/src/bonsai/bonsai/bim/module/root/operator.py +++ b/src/bonsai/bonsai/bim/module/root/operator.py @@ -43,7 +43,7 @@ class EnableReassignClass(bpy.types.Operator): def execute(self, context): obj = context.active_object - self.file = IfcStore.get_file() + self.file = tool.Ifc.get() ifc_class = obj.name.split("/")[0] context.active_object.BIMObjectProperties.is_reassigning_class = True ifc_products = [ @@ -93,7 +93,7 @@ class ReassignClass(bpy.types.Operator, tool.Ifc.Operator): objects = [bpy.data.objects.get(self.obj)] else: objects = set(context.selected_objects + [context.active_object]) - self.file = IfcStore.get_file() + self.file = tool.Ifc.get() root_props = context.scene.BIMRootProperties ifc_product: str = root_props.ifc_product ifc_class: str = root_props.ifc_class diff --git a/src/bonsai/bonsai/bim/module/root/prop.py b/src/bonsai/bonsai/bim/module/root/prop.py index bcc0b69456..9e4e4ab1fc 100644 --- a/src/bonsai/bonsai/bim/module/root/prop.py +++ b/src/bonsai/bonsai/bim/module/root/prop.py @@ -20,6 +20,7 @@ import bpy import ifcopenshell import ifcopenshell.util.element import ifcopenshell.util.schema +import ifcopenshell.util.type import bonsai.tool as tool from bonsai.bim.module.root.data import IfcClassData from bpy.types import PropertyGroup diff --git a/src/bonsai/bonsai/tool/pset.py b/src/bonsai/bonsai/tool/pset.py index 48006aa270..bcac21d7ca 100644 --- a/src/bonsai/bonsai/tool/pset.py +++ b/src/bonsai/bonsai/tool/pset.py @@ -22,6 +22,7 @@ import ifcopenshell import ifcopenshell.api.pset import ifcopenshell.util.attribute import ifcopenshell.util.element +import bonsai.bim.helper import bonsai.core.tool import bonsai.tool as tool import bonsai.bim.schema