Minor fix for previous commit

This commit is contained in:
Dion Moult
2021-02-21 11:55:04 +11:00
parent 0286cd370d
commit 25d86a50c5
3 changed files with 17 additions and 9 deletions
@@ -160,8 +160,12 @@ class AssignClass(bpy.types.Operator):
for collection in obj.users_collection:
if "Ifc" not in collection.name or collection.name == obj.name:
continue
bpy.ops.bim.assign_container(relating_structure=collection.name, related_element=obj.name)
break
spatial_obj = bpy.data.objects.get(collection.name)
if spatial_obj and spatial_obj.BIMObjectProperties.ifc_definition_id:
bpy.ops.bim.assign_container(
relating_structure=spatial_obj.BIMObjectProperties.ifc_definition_id, related_element=obj.name
)
break
class UnassignClass(bpy.types.Operator):