From 8a55f18c9506b8e55a3a442771957e899f683b93 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Mon, 19 Aug 2024 17:03:32 +0500 Subject: [PATCH] bonsai bcf - poll message for adding bim snippet --- src/bonsai/bonsai/bim/module/bcf/operator.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/bonsai/bonsai/bim/module/bcf/operator.py b/src/bonsai/bonsai/bim/module/bcf/operator.py index 2fbe559115..cc758cb6e6 100644 --- a/src/bonsai/bonsai/bim/module/bcf/operator.py +++ b/src/bonsai/bonsai/bim/module/bcf/operator.py @@ -379,12 +379,14 @@ class AddBcfBimSnippet(bpy.types.Operator): @classmethod def poll(cls, context): - return all( - ( - getattr(context.scene.BCFProperties, attr, False) - for attr in ("bim_snippet_reference", "bim_snippet_schema", "bim_snippet_type") - ) + props = context.scene.BCFProperties + props_are_filled = all( + (getattr(props, attr) for attr in ("bim_snippet_reference", "bim_snippet_schema", "bim_snippet_type")) ) + if not props_are_filled: + cls.poll_message_set("Some BIM snippet fields are empty.") + return False + return True def execute(self, context): bcfxml = bcfstore.BcfStore.get_bcfxml()