From 9671be173409b7d5e94b55eda141364dbf99d4b4 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Thu, 8 Aug 2024 18:10:14 +0500 Subject: [PATCH] add a couple poll messages --- src/blenderbim/blenderbim/bim/module/cost/operator.py | 7 ++----- src/blenderbim/blenderbim/bim/module/sequence/operator.py | 7 ++----- 2 files changed, 4 insertions(+), 10 deletions(-) diff --git a/src/blenderbim/blenderbim/bim/module/cost/operator.py b/src/blenderbim/blenderbim/bim/module/cost/operator.py index 3b384cf60c..af76c93e2f 100644 --- a/src/blenderbim/blenderbim/bim/module/cost/operator.py +++ b/src/blenderbim/blenderbim/bim/module/cost/operator.py @@ -720,11 +720,8 @@ class LoadProductCostItems(bpy.types.Operator): @classmethod def poll(cls, context): - if not tool.Ifc.get(): - return False - if not context.active_object: - return False - if not context.active_object.BIMObjectProperties.ifc_definition_id: + if not tool.Ifc.get() or not (obj := context.active_object) or not (obj.BIMObjectProperties.ifc_definition_id): + cls.poll_message_set("No IFC object is active.") return False return True diff --git a/src/blenderbim/blenderbim/bim/module/sequence/operator.py b/src/blenderbim/blenderbim/bim/module/sequence/operator.py index d64b0400ed..6dd6f47c09 100644 --- a/src/blenderbim/blenderbim/bim/module/sequence/operator.py +++ b/src/blenderbim/blenderbim/bim/module/sequence/operator.py @@ -1399,11 +1399,8 @@ class LoadProductTasks(bpy.types.Operator): @classmethod def poll(cls, context): - if not tool.Ifc.get(): - return False - if not context.active_object: - return False - if not context.active_object.BIMObjectProperties.ifc_definition_id: + if not tool.Ifc.get() or not (obj := context.active_object) or not (obj.BIMObjectProperties.ifc_definition_id): + cls.poll_message_set("No IFC object is active.") return False return True