sequence.copy_work_schedule #6901

Similar to e50215ddc but for IfcWorkSchedules

Button location - https://files.catbox.moe/jfd1ol.png

src/ifcopenshell-python/ifcopenshell/api/sequence/copy_work_schedule.py

@@ -0,0 +1,51 @@
+# IfcOpenShell - IFC toolkit and geometry engine
+# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
+#
+# This file is part of IfcOpenShell.
+#
+# IfcOpenShell is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# IfcOpenShell is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with IfcOpenShell.  If not, see <http://www.gnu.org/licenses/>.
+
+import ifcopenshell.api.control
+import ifcopenshell.api.sequence
+import ifcopenshell.util.element
+
+
+def copy_work_schedule(
+    file: ifcopenshell.file,
+    work_schedule: ifcopenshell.entity_instance,
+) -> ifcopenshell.entity_instance:
+    """Copy a work schedule.
+
+    :param work_schedule: IfcWorkSchedule to copy.
+    :return: The duplicated IfcWorkSchedule entity.
+
+    Example:
+
+    .. code:: python
+
+        work_plan = ifcopenshell.api.sequence.add_work_plan(model, name="Construction")
+        schedule = ifcopenshell.api.sequence.add_work_schedule(model,
+            name="Construction Schedule A", work_plan=work_plan)
+        new_schedule = ifcopenshell.api.sequence.copy_work_schedule(model, schedule)
+    """
+    # Shared code logic with copy_cost_schedule.
+    new_schedule = ifcopenshell.util.element.copy(file, work_schedule)
+
+    for rel in work_schedule.Controls:
+        for task in rel.RelatedObjects:
+            duplicated_tasks = ifcopenshell.api.sequence.duplicate_task(file, task)[1]
+            # All other nested items are not connected to the work schedule explicitly.
+            duplicated_task = duplicated_tasks[0]
+            ifcopenshell.api.control.assign_control(file, new_schedule, duplicated_task)
+    return new_schedule