2021-01-21 21:58:27 +11:00
|
|
|
class Usecase:
|
2021-03-25 00:54:42 +01:00
|
|
|
def __init__(self, file, settings={}):
|
2021-01-21 21:58:27 +11:00
|
|
|
self.file = file
|
|
|
|
|
self.settings = {"classification": None}
|
|
|
|
|
for key, value in settings.items():
|
|
|
|
|
self.settings[key] = value
|
|
|
|
|
|
|
|
|
|
def execute(self):
|
|
|
|
|
references = self.get_references(self.settings["classification"])
|
|
|
|
|
for reference in references:
|
|
|
|
|
self.file.remove(reference)
|
|
|
|
|
self.file.remove(self.settings["classification"])
|
|
|
|
|
for rel in self.file.by_type("IfcRelAssociatesClassification"):
|
|
|
|
|
if not rel.RelatingClassification:
|
|
|
|
|
self.file.remove(rel)
|
|
|
|
|
|
|
|
|
|
def get_references(self, classification):
|
|
|
|
|
results = []
|
|
|
|
|
if not classification.HasReferences:
|
|
|
|
|
return results
|
|
|
|
|
for reference in classification.HasReferences:
|
|
|
|
|
results.append(reference)
|
|
|
|
|
results.extend(self.get_references(reference))
|
|
|
|
|
return results
|