Removing an organisation now properly handles references appropriately.

This commit is contained in:
Dion Moult
2021-10-01 13:29:10 +10:00
parent 915fa9f9b0
commit 291bb0cd55
3 changed files with 128 additions and 0 deletions
@@ -0,0 +1,12 @@
import ifcopenshell.api
class Usecase:
def __init__(self, file, **settings):
self.file = file
self.settings = {"application": None}
for key, value in settings.items():
self.settings[key] = value
def execute(self):
self.file.remove(self.settings["application"])
@@ -1,3 +1,6 @@
import ifcopenshell.api
class Usecase:
def __init__(self, file, **settings):
self.file = file
@@ -6,4 +9,29 @@ class Usecase:
self.settings[key] = value
def execute(self):
for role in self.settings["organisation"].Roles or []:
if len(self.file.get_inverse(role)) == 1:
ifcopenshell.api.run("owner.remove_role", self.file, role=role)
for address in self.settings["organisation"].Addresses or []:
if len(self.file.get_inverse(address)) == 1:
ifcopenshell.api.run("owner.remove_address", self.file, address=address)
for inverse in self.file.get_inverse(self.settings["organisation"]):
if inverse.is_a("IfcOrganizationRelationship"):
if inverse.RelatingOrganization == self.settings["organisation"]:
self.file.remove(inverse)
elif inverse.RelatedOrganizations == (self.settings["organisation"],):
self.file.remove(inverse)
elif inverse.is_a("IfcDocumentInformation"):
if inverse.Editors == (self.settings["organisation"],):
inverse.Editors = None
elif inverse.is_a("IfcPersonAndOrganization"):
ifcopenshell.api.run("owner.remove_person_and_organisation", self.file, person_and_organisation=inverse)
elif inverse.is_a("IfcActor"):
ifcopenshell.api.run("owner.remove_actor", self.file, actor=inverse)
elif inverse.is_a("IfcResourceLevelRelationship") and not inverse.is_a("IfcOrganizationRelationship"):
if inverse.RelatedResourceObjects == (self.settings["organisation"],):
self.file.remove(inverse)
elif inverse.is_a("IfcApplication"):
ifcopenshell.api.run("owner.remove_application", self.file, application=inverse)
self.file.remove(self.settings["organisation"])