From a374ee39d36b2a6e5ce277034e40c8a2d1878030 Mon Sep 17 00:00:00 2001 From: "Sigma Dimensions (Yass)" <79010126+myoualid@users.noreply.github.com> Date: Fri, 11 Aug 2023 02:06:49 +0100 Subject: [PATCH] simplify setup_default_task_columns --- .../blenderbim/bim/module/sequence/ui.py | 2 +- src/blenderbim/blenderbim/tool/sequence.py | 20 +++---------------- 2 files changed, 4 insertions(+), 18 deletions(-) diff --git a/src/blenderbim/blenderbim/bim/module/sequence/ui.py b/src/blenderbim/blenderbim/bim/module/sequence/ui.py index b945b72e4b..43cbb3bf70 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/ui.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/ui.py @@ -233,7 +233,7 @@ class BIM_PT_work_schedules(Panel): if not self.props.should_show_column_ui: return row = self.layout.row() - row.operator("bim.setup_default_task_columns", text="Add Default Columns", icon="ANCHOR_BOTTOM") + row.operator("bim.setup_default_task_columns", text="Setup Default Columns", icon="ANCHOR_BOTTOM") row.alignment = "RIGHT" row = self.layout.row(align=True) row.prop(self.props, "column_types", text="") diff --git a/src/blenderbim/blenderbim/tool/sequence.py b/src/blenderbim/blenderbim/tool/sequence.py index 157f2a6df7..6c62dd2c5f 100644 --- a/src/blenderbim/blenderbim/tool/sequence.py +++ b/src/blenderbim/blenderbim/tool/sequence.py @@ -719,26 +719,12 @@ class Sequence(blenderbim.core.tool.Sequence): @classmethod def setup_default_task_columns(cls): - items = [ - { - "column_type": "IfcTaskTime", - "name": "ScheduleStart", - }, - { - "column_type": "IfcTaskTime", - "name": "ScheduleFinish", - }, - { - "column_type": "IfcTaskTime", - "name": "ScheduleDuration", - }, - ] - props = bpy.context.scene.BIMWorkScheduleProperties props.columns.clear() - for item in items: + default_columns = ["ScheduleStart","ScheduleFinish","ScheduleDuration"] + for item in default_columns: new = props.columns.add() - new.name = f"{item['column_type']}.{item['name']}" + new.name = f"IfcTaskTime.{item}" new.data_type = "string" @classmethod