From ebfaeafeeebef2d1aabf57680361a640d3d1d2b7 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Tue, 9 Mar 2021 14:25:04 +1100
Subject: [PATCH] You can now bulk assign / unassign construction types

---
 .../blenderbim/bim/module/type/operator.py    | 52 ++++++++++---------
 1 file changed, 28 insertions(+), 24 deletions(-)

diff --git a/src/ifcblenderexport/blenderbim/bim/module/type/operator.py b/src/ifcblenderexport/blenderbim/bim/module/type/operator.py
index 342bfd1c05..343d13ee35 100644
--- a/src/ifcblenderexport/blenderbim/bim/module/type/operator.py
+++ b/src/ifcblenderexport/blenderbim/bim/module/type/operator.py
@@ -18,21 +18,22 @@ class AssignType(bpy.types.Operator):
 
     def execute(self, context):
         self.file = IfcStore.get_file()
-        related_object = bpy.data.objects.get(self.related_object) if self.related_object else bpy.context.active_object
-        oprops = related_object.BIMObjectProperties
-        props = related_object.BIMTypeProperties
-        relating_type = self.relating_type or int(props.relating_type)
-        assign_type.Usecase(
-            self.file,
-            {
-                "related_object": self.file.by_id(oprops.ifc_definition_id),
-                "relating_type": self.file.by_id(relating_type),
-            },
-        ).execute()
-        Data.load(oprops.ifc_definition_id)
-
-        if self.file.by_id(relating_type).RepresentationMaps:
-            bpy.ops.bim.map_representations(product_id=oprops.ifc_definition_id, type_product_id=relating_type)
+        relating_type = self.relating_type or int(context.active_object.BIMTypeProperties.relating_type)
+        related_objects = (
+            [bpy.data.objects.get(self.related_object)] if self.related_object else bpy.context.selected_objects
+        )
+        for related_object in related_objects:
+            oprops = related_object.BIMObjectProperties
+            assign_type.Usecase(
+                self.file,
+                {
+                    "related_object": self.file.by_id(oprops.ifc_definition_id),
+                    "relating_type": self.file.by_id(relating_type),
+                },
+            ).execute()
+            Data.load(oprops.ifc_definition_id)
+            if self.file.by_id(relating_type).RepresentationMaps:
+                bpy.ops.bim.map_representations(product_id=oprops.ifc_definition_id, type_product_id=relating_type)
 
         bpy.ops.bim.disable_editing_type(obj=related_object.name)
         return {"FINISHED"}
@@ -45,15 +46,18 @@ class UnassignType(bpy.types.Operator):
 
     def execute(self, context):
         self.file = IfcStore.get_file()
-        related_object = bpy.data.objects.get(self.related_object) if self.related_object else bpy.context.active_object
-        oprops = related_object.BIMObjectProperties
-        unassign_type.Usecase(
-            self.file,
-            {
-                "related_object": self.file.by_id(oprops.ifc_definition_id),
-            },
-        ).execute()
-        Data.load(oprops.ifc_definition_id)
+        related_objects = (
+            [bpy.data.objects.get(self.related_object)] if self.related_object else bpy.context.selected_objects
+        )
+        for related_object in related_objects:
+            oprops = related_object.BIMObjectProperties
+            unassign_type.Usecase(
+                self.file,
+                {
+                    "related_object": self.file.by_id(oprops.ifc_definition_id),
+                },
+            ).execute()
+            Data.load(oprops.ifc_definition_id)
         return {"FINISHED"}