Implement unassigning products to tasks

This commit is contained in:
Dion Moult
2021-04-21 11:53:33 +10:00
parent 3143f8c9a5
commit 9b855bed2d
6 changed files with 69 additions and 6 deletions
@@ -1,6 +1,7 @@
import ifcopenshell.api
import ifcopenshell
class Usecase:
def __init__(self, file, **settings):
self.file = file
@@ -16,9 +17,9 @@ class Usecase:
"root.create_entity",
self.file,
ifc_class="IfcTask",
name= None,
predefined_type= "NOTDEFINED",
identification= "none",
name=None,
predefined_type="NOTDEFINED",
identification="none",
)
task.IsMilestone = False
if self.settings["work_schedule"]:
@@ -33,6 +34,6 @@ class Usecase:
)
elif self.settings["parent_task"]:
ifcopenshell.api.run(
"nest.assign_object", self.file, object=task, relating_object=self.settings["parent_task"]
"nest.assign_object", self.file, related_object=task, relating_object=self.settings["parent_task"]
)
return task
@@ -79,12 +79,18 @@ class Data:
data = task.get_info()
del data["OwnerHistory"]
data["RelatedObjects"] = []
data["RelatingProducts"] = []
data["IsPredecessorTo"] = []
data["IsSuccessorFrom"] = []
if task.TaskTime:
data["TaskTime"] = data["TaskTime"].id()
for rel in task.IsNestedBy:
[data["RelatedObjects"].append(o.id()) for o in rel.RelatedObjects if o.is_a("IfcTask")]
[
data["RelatingProducts"].append(r.RelatingProduct.id())
for r in task.HasAssignments
if r.is_a("IfcRelAssignsToProduct")
]
[data["IsPredecessorTo"].append(rel.RelatedProcess.id()) for rel in task.IsPredecessorTo or []]
[data["IsSuccessorFrom"].append(rel.RelatingProcess.id()) for rel in task.IsSuccessorFrom or []]
cls.tasks[task.id()] = data
@@ -0,0 +1,25 @@
import ifcopenshell
import ifcopenshell.api
class Usecase:
def __init__(self, file, **settings):
self.file = file
self.settings = {
"relating_product": None,
"related_object": None,
}
for key, value in settings.items():
self.settings[key] = value
def execute(self):
for rel in self.settings["related_object"].HasAssignments or []:
if not rel.is_a("IfcRelAssignsToProduct") or rel.RelatingProduct != self.settings["relating_product"]:
continue
if len(rel.RelatedObjects) == 1:
return self.file.remove(rel)
related_objects = list(rel.RelatedObjects)
related_objects.remove(self.settings["related_object"])
rel.RelatedObjects = related_objects
ifcopenshell.api.run("owner.update_owner_history", self.file, **{"element": rel})
return rel