mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 17:58:20 +00:00
New assign / unassign declaration usecase, minor code & formatting cleanup
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user