Fix bug where removing objects could leave orphaned annotation relationships

This commit is contained in:
Dion Moult
2023-03-19 15:17:26 +11:00
parent 558aff72a0
commit 3da718a6d1
2 changed files with 12 additions and 0 deletions
@@ -130,4 +130,9 @@ class Usecase:
elif inverse.is_a("IfcRelAssignsToGroup"):
if len(inverse.RelatedObjects) == 1:
self.file.remove(inverse)
elif inverse.is_a("IfcRelAssignsToProduct"):
if inverse.RelatingProduct == self.settings["product"]:
self.file.remove(inverse)
elif len(inverse.RelatedObject) == 1:
self.file.remove(inverse)
self.file.remove(self.settings["product"])
@@ -245,3 +245,10 @@ class TestRemoveProduct(test.bootstrap.IFC4):
ifcopenshell.api.run("group.assign_group", self.file, products=[element], group=group)
ifcopenshell.api.run("root.remove_product", self.file, product=element)
assert not self.file.by_type("IfcRelAssignsToGroup")
def test_removing_product_assignments(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
annotation = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcAnnotation")
ifcopenshell.api.run("drawing.assign_product", self.file, relating_product=element, related_object=annotation)
ifcopenshell.api.run("root.remove_product", self.file, product=element)
assert not self.file.by_type("IfcRelAssignsToProduct")