From 828160c1b1f92d5b7272b091e48fb692183724bf Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Fri, 18 Oct 2024 15:29:45 +0500 Subject: [PATCH] Fix UI errors if there are no pset template files --- src/bonsai/bonsai/bim/module/pset_template/data.py | 4 +++- src/bonsai/bonsai/bim/module/pset_template/ui.py | 5 ++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/src/bonsai/bonsai/bim/module/pset_template/data.py b/src/bonsai/bonsai/bim/module/pset_template/data.py index ce9917f38b..76ceaa18fa 100644 --- a/src/bonsai/bonsai/bim/module/pset_template/data.py +++ b/src/bonsai/bonsai/bim/module/pset_template/data.py @@ -87,7 +87,9 @@ class PsetTemplatesData: return sorted(results, key=lambda x: x[1]) @classmethod - def pset_templates(cls): + def pset_templates(cls) -> list[tuple[str, str, str]]: + if not cls.data["pset_template_files"]: + return [] if not IfcStore.pset_template_file: IfcStore.pset_template_path = bpy.context.scene.BIMPsetTemplateProperties.pset_template_files IfcStore.pset_template_file = ifcopenshell.open(IfcStore.pset_template_path) diff --git a/src/bonsai/bonsai/bim/module/pset_template/ui.py b/src/bonsai/bonsai/bim/module/pset_template/ui.py index ef9870338f..47020fc712 100644 --- a/src/bonsai/bonsai/bim/module/pset_template/ui.py +++ b/src/bonsai/bonsai/bim/module/pset_template/ui.py @@ -38,7 +38,10 @@ class BIM_PT_pset_template(Panel): self.props = context.scene.BIMPsetTemplateProperties row = self.layout.row(align=True) - prop_with_search(row, self.props, "pset_template_files", text="", icon="FILE") + if PsetTemplatesData.data["pset_template_files"]: + prop_with_search(row, self.props, "pset_template_files", text="", icon="FILE") + else: + row.label(text="No Pset Template Files", icon="FILE") row.operator("bim.add_pset_template_file", icon="ADD", text="") row.operator("bim.remove_pset_template_file", icon="X", text="")