sequence.remove_work_calendar to remove related work times

Previously they were leaved orphaned
This commit is contained in:
Andrej730
2024-06-10 12:22:21 +05:00
parent 7c713eee65
commit 1343741cfd
3 changed files with 66 additions and 1 deletions
@@ -16,10 +16,13 @@
# 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
from typing import Literal
TIME_TYPE = Literal["WorkingTimes", "ExceptionTimes"]
def add_work_time(
file: ifcopenshell.file, work_calendar: ifcopenshell.entity_instance, time_type: str = "WorkingTimes"
file: ifcopenshell.file, work_calendar: ifcopenshell.entity_instance, time_type: TIME_TYPE = "WorkingTimes"
) -> ifcopenshell.entity_instance:
"""Add either working times or holiday times to a calendar
@@ -18,6 +18,7 @@
import ifcopenshell
import ifcopenshell.api
import ifcopenshell.api.sequence
import ifcopenshell.util.element
@@ -60,6 +61,15 @@ def remove_work_calendar(file: ifcopenshell.file, work_calendar: ifcopenshell.en
relating_control=settings["work_calendar"],
related_object=related_object,
)
# Currently in API work times are created already attached
# to the work calendar, so they are never reused.
for working_time in settings["work_calendar"].WorkingTimes or []:
ifcopenshell.api.sequence.remove_work_time(file, work_time=working_time)
for exception_time in settings["work_calendar"].ExceptionTimes or []:
ifcopenshell.api.sequence.remove_work_time(file, work_time=exception_time)
history = settings["work_calendar"].OwnerHistory
file.remove(settings["work_calendar"])
if history: