New assign / unassign declaration usecase, minor code & formatting cleanup

This commit is contained in:
Dion Moult
2021-04-07 13:15:16 +10:00
parent 13d4a99358
commit ae0789e4ec
25 changed files with 289 additions and 228 deletions
@@ -1,31 +1,26 @@
import ifcopenshell
import ifcopenshell.api
class Usecase:
def __init__(self, file, **settings):
self.file = file
self.settings = {
"work_schedule": None,
"work_plan":None,
}
self.settings = {"work_schedule": None, "work_plan": None}
for key, value in settings.items():
self.settings[key] = value
def execute(self):
###Project declaration should be removed if the workschedule is assigned to a project.####
if self.settings["work_schedule"].HasContext:
rel_declares = self.settings["work_schedule"].HasContext[0]
related_definitions = list(rel_declares.RelatedDefinitions)
related_definitions.remove(self.settings["work_schedule"])
rel_declares.RelatedDefinitions = related_definitions
if self.settings["work_plan"].IsDecomposedBy:
rel_aggregates = self.settings["work_plan"].IsDecomposedBy[0]
related_objects = list(rel_aggregates.RelatedObjects or [])
related_objects.append(self.settings["work_schedule"])
rel_aggregates.RelatedObjects = related_objects
else:
rel_aggregates = ifcopenshell.api.run("aggregate.assign_object", self.file, **{
"product": self.settings["work_schedule"],
"relating_object": self.settings["work_plan"]
})
return rel_aggregates
# TODO: this is an ambiguity by buildingSMART
# See https://forums.buildingsmart.org/t/is-the-ifcworkschedule-project-declaration-mutually-exclusive-to-aggregation-within-a-relating-ifcworkplan/3510
ifcopenshell.api.run(
"project.unassign_declaration",
self.file,
definition=self.settings["work_schedule"],
relating_context=self.file.by_type("IfcContext")[0],
)
rel_aggregates = ifcopenshell.api.run(
"aggregate.assign_object",
self.file,
**{"product": self.settings["work_schedule"], "relating_object": self.settings["work_plan"]}
)
return rel_aggregates