mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-19 06:39:13 +00:00
Fix bug where adding heavy classification references led to duplicate classification systems
This commit is contained in:
@@ -58,13 +58,30 @@ class Usecase:
|
|||||||
return
|
return
|
||||||
|
|
||||||
migrator = ifcopenshell.util.schema.Migrator()
|
migrator = ifcopenshell.util.schema.Migrator()
|
||||||
|
|
||||||
if self.settings["is_lightweight"]:
|
if self.settings["is_lightweight"]:
|
||||||
old_referenced_source = self.settings["reference"].ReferencedSource
|
old_referenced_source = self.settings["reference"].ReferencedSource
|
||||||
self.settings["reference"].ReferencedSource = None
|
self.settings["reference"].ReferencedSource = None
|
||||||
|
else:
|
||||||
|
existing_classification = [
|
||||||
|
c for c in self.file.by_type("IfcClassification") if c.Name == self.settings["classification"].Name
|
||||||
|
]
|
||||||
|
|
||||||
relating_classification = migrator.migrate(self.settings["reference"], self.file)
|
relating_classification = migrator.migrate(self.settings["reference"], self.file)
|
||||||
|
|
||||||
if self.settings["is_lightweight"]:
|
if self.settings["is_lightweight"]:
|
||||||
relating_classification.ReferencedSource = self.settings["classification"]
|
relating_classification.ReferencedSource = self.settings["classification"]
|
||||||
self.settings["reference"].ReferencedSource = old_referenced_source
|
self.settings["reference"].ReferencedSource = old_referenced_source
|
||||||
|
elif existing_classification:
|
||||||
|
to_delete = set()
|
||||||
|
for traversed_reference in self.file.traverse(relating_classification):
|
||||||
|
if traversed_reference.ReferencedSource.is_a("IfcClassification"):
|
||||||
|
to_delete.add(traversed_reference.ReferencedSource)
|
||||||
|
traversed_reference.ReferencedSource = existing_classification[0]
|
||||||
|
break
|
||||||
|
for element in to_delete:
|
||||||
|
self.file.remove(element)
|
||||||
|
|
||||||
self.file.create_entity(
|
self.file.create_entity(
|
||||||
"IfcRelAssociatesClassification",
|
"IfcRelAssociatesClassification",
|
||||||
**{
|
**{
|
||||||
|
|||||||
Reference in New Issue
Block a user