Code review for assign product with myoualid

This commit is contained in:
Dion Moult
2021-04-21 11:11:08 +10:00
parent 8dd781aea2
commit 3143f8c9a5
6 changed files with 52 additions and 79 deletions
@@ -7,50 +7,37 @@ class Usecase:
self.file = file
self.settings = {
"relating_product": None,
"related_process": None,
"related_object": None,
}
for key, value in settings.items():
self.settings[key] = value
def execute(self):
if self.settings["related_object"].HasAssignments:
for assignment in self.settings["related_object"].HasAssignments:
if (
assignment.is_a("IfclRelAssignsToProduct")
and assignment.RelatingProduct == self.settings["relating_product"]
):
return
referenced_by = None
if self.settings["relating_product"].ReferencedBy:
for rel in self.settings["relating_product"].ReferencedBy:
if rel.is_a("IfcRelAssignsToProduct"):
referenced_by = rel
assignment = None
for rel in self.settings["related_process"].HasAssignments:
if rel.is_a('IfcRelAssignsToProduct'):
assignment = rel
break
if referenced_by and referenced_by == assignment:
return
if assignment:
related_objects = set(assignment.RelatedObjects)
related_objects.add(self.settings["relating_product"])
assignment.RelatedObjects = list(related_objects)
ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": assignment})
else:
rel = self.file.create_entity(
"IfcRelAssignsToProduct",
**{
"GlobalId": ifcopenshell.guid.new(),
"OwnerHistory": ifcopenshell.api.run("owner.create_owner_history", self.file),
"RelatingProduct": self.settings["relating_product"],
"RelatedObjects": [self.settings["related_process"]],
})
return rel
referenced_by = self.settings["relating_product"].ReferencedBy[0]
if referenced_by:
related_objects = list(referenced_by.RelatedObjects)
related_objects.remove(self.settings["relating_product"])
if related_objects:
referenced_by.RelatedObjects = related_objects
ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": referenced_by})
else:
self.file.remove(referenced_by)
return rel
related_objects.append(self.settings["related_object"])
referenced_by.RelatedObjects = related_objects
ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": referenced_by})
else:
referenced_by = self.file.create_entity(
"IfcRelAssignsToProduct",
**{
"GlobalId": ifcopenshell.guid.new(),
"OwnerHistory": ifcopenshell.api.run("owner.create_owner_history", self.file),
"RelatedObjects": [self.settings["related_object"]],
"RelatingProduct": self.settings["relating_product"],
}
)
return referenced_by